CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-26 18:20:26 +12:00
Code Issues Releases Wiki Activity

Fix tools/cudnn2cunn.lua no effect

Browse source
This commit is contained in:
nagadomi 2015-11-06 10:32:13 +09:00
parent 903d945652
commit 6924920a2f
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tools/cudnn2cunn.lua
View file

@ -9,8 +9,8 @@ require 'cudnn'
require 'w2nn'
local srcnn = require 'srcnn'
local function cudnn2cunn(cunn_model)
local cudnn_model = srcnn.waifu2x_cunn(srcnn.channels(cunn_model))
local function cudnn2cunn(cudnn_model)
local cunn_model = srcnn.waifu2x_cunn(srcnn.channels(cudnn_model))
local weight_from = cudnn_model:findModules("cudnn.SpatialConvolution")
local weight_to = cunn_model:findModules("nn.SpatialConvolutionMM")