8dea362bed
- Memory compression by snappy (lua-csnappy) - Use RGB-wise Weighted MSE(R*0.299, G*0.587, B*0.114) instead of MSE - Aggressive cropping for edge region and some change.
20 lines
367 B
Lua
20 lines
367 B
Lua
local function load_cunn()
|
|
require 'nn'
|
|
require 'cunn'
|
|
end
|
|
local function load_cudnn()
|
|
require 'cudnn'
|
|
cudnn.fastest = true
|
|
end
|
|
if mynn then
|
|
return mynn
|
|
else
|
|
load_cunn()
|
|
--load_cudnn()
|
|
mynn = {}
|
|
require './LeakyReLU'
|
|
require './LeakyReLU_deprecated'
|
|
require './DepthExpand2x'
|
|
require './RGBWeightedMSECriterion'
|
|
return mynn
|
|
end
|