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

nobias

Browse source
This commit is contained in:
nagadomi 2018-10-27 06:05:10 +00:00
parent 0b82668359
commit 8051535fdc
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/srcnn.lua
View file

@ -845,9 +845,9 @@ function srcnn.upcunet_v2(backend, ch)
model:add(unet_conv(in_ch, 32, 64, false))
model:add(unet_branch(block2, backend, 64, 64, 16))
if deconv then
model:add(SpatialFullConvolution(backend, 64, out_ch, 4, 4, 2, 2, 3, 3))
model:add(SpatialFullConvolution(backend, 64, out_ch, 4, 4, 2, 2, 3, 3):noBias())
else
model:add(SpatialConvolution(backend, 64, out_ch, 3, 3, 1, 1, 0, 0))
model:add(SpatialConvolution(backend, 64, out_ch, 3, 3, 1, 1, 0, 0):noBias())
end
return model
end