29 lines
608 B
Lua
29 lines
608 B
Lua
local function load_nn()
|
|
require 'torch'
|
|
require 'nn'
|
|
end
|
|
local function load_cunn()
|
|
require 'cutorch'
|
|
require 'cunn'
|
|
end
|
|
local function load_cudnn()
|
|
require 'cudnn'
|
|
cudnn.benchmark = true
|
|
end
|
|
if w2nn then
|
|
return w2nn
|
|
else
|
|
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 = {}
|
|
require 'LeakyReLU'
|
|
require 'LeakyReLU_deprecated'
|
|
require 'DepthExpand2x'
|
|
require 'PSNRCriterion'
|
|
require 'ClippedWeightedHuberCriterion'
|
|
return w2nn
|
|
end
|