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
24 lines
705 B
Lua
24 lines
705 B
Lua
-- adapted from https://github.com/marcan/cl-waifu2x
|
|
require './lib/portable'
|
|
require './lib/LeakyReLU'
|
|
local cjson = require "cjson"
|
|
|
|
local model = torch.load(arg[1], "ascii")
|
|
|
|
local jmodules = {}
|
|
local modules = model:findModules("nn.SpatialConvolutionMM")
|
|
for i = 1, #modules, 1 do
|
|
local module = modules[i]
|
|
local jmod = {
|
|
kW = module.kW,
|
|
kH = module.kH,
|
|
nInputPlane = module.nInputPlane,
|
|
nOutputPlane = module.nOutputPlane,
|
|
bias = torch.totable(module.bias:float()),
|
|
weight = torch.totable(module.weight:float():reshape(module.nOutputPlane, module.nInputPlane, module.kW, module.kH))
|
|
}
|
|
table.insert(jmodules, jmod)
|
|
end
|
|
|
|
io.write(cjson.encode(jmodules))
|