CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-20 04:42:19 +12:00
Code Issues Releases Wiki Activity

no confidence change for #128

Browse source
This commit is contained in:
nagadomi 2016-07-27 22:55:56 +09:00
parent 8739f5487f
commit 382d493514
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
lib/w2nn.lua
View file

@ -7,18 +7,17 @@ local function load_cunn()
require 'cunn'
end
local function load_cudnn()
require 'cudnn'
cudnn.benchmark = true
cudnn = require('cudnn')
end
if w2nn then
return w2nn
else
w2nn = {}
local state, ret = pcall(load_cunn)
if not state then
error("Failed to load CUDA modules. Please check the CUDA Settings.\n---\n" .. ret)
end
pcall(load_cudnn)
w2nn = {}
function w2nn.load_model(model_path, force_cudnn)
local model = torch.load(model_path, "ascii")