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.
22 lines
457 B
Lua
22 lines
457 B
Lua
local function load_cuda()
|
|
require 'nn'
|
|
require 'cunn'
|
|
end
|
|
local function load_cudnn()
|
|
require 'cudnn'
|
|
--cudnn.fastest = true
|
|
end
|
|
|
|
if pcall(load_cuda) then
|
|
else
|
|
--[[ TODO: fakecuda does not work.
|
|
|
|
io.stderr:write("use FakeCUDA; if you have NVIDIA GPU, Please install cutorch and cunn. FakeCuda will be extremely slow.\n")
|
|
require 'torch'
|
|
require 'nn'
|
|
require('fakecuda').init(true)
|
|
--]]
|
|
end
|
|
if pcall(load_cudnn) then
|
|
end
|