CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00
Code Issues Releases Wiki Activity

Fix shortcut connection order in unet_branch

Browse source
This commit is contained in:
nagadomi 2018-11-01 20:00:11 +09:00
parent 2e59d97b17
commit 3654ee41cf
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib/srcnn.lua
View file

@ -568,8 +568,8 @@ local function unet_branch(backend, insert, backend, n_input, n_output, depad)
block:add(insert)
block:add(SpatialFullConvolution(backend, n_output, n_output, 2, 2, 2, 2, 0, 0))-- upsampling
block:add(nn.LeakyReLU(0.1, true))
con:add(nn.SpatialZeroPadding(-depad, -depad, -depad, -depad))
con:add(block)
con:add(nn.SpatialZeroPadding(-depad, -depad, -depad, -depad))
model:add(con)
model:add(nn.CAddTable())
return model