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