a706892b59
- remove support for cuDNN - add new pre-trained model and json files - some changes in training script If you have cuDNN model, Please run following commands to convert. $ cp models/your_own_model.t7 models/your_own_model.t7.backup $ th cudnn2cunn.lua -model models/your_own_model.t7
16 lines
352 B
Lua
16 lines
352 B
Lua
local function load_cuda()
|
|
require 'cunn'
|
|
end
|
|
|
|
if pcall(load_cuda) then
|
|
require 'cunn'
|
|
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
|