waifu2x/lib/w2nn.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
   pcall(load_cunn)
   pcall(load_cudnn)
   w2nn = {}
   require 'LeakyReLU'
   require 'LeakyReLU_deprecated'
   require 'DepthExpand2x'
   require 'WeightedMSECriterion'
   require 'ClippedWeightedHuberCriterion'
   require 'cleanup_model'
   return w2nn
end