Explorar o código

Fix srresnet_2x

nagadomi %!s(int64=8) %!d(string=hai) anos
pai
achega
4be1141292
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      lib/srcnn.lua

+ 1 - 0
lib/srcnn.lua

@@ -403,6 +403,7 @@ function srcnn.srresnet_2x(backend, ch)
       conv:add(SpatialConvolution(backend, 64, 64, 3, 3, 1, 1, 0, 0))
       conv:add(ReLU(backend))
       conv:add(SpatialConvolution(backend, 64, 64, 3, 3, 1, 1, 0, 0))
+      conv:add(ReLU(backend))
       con:add(conv)
       con:add(nn.SpatialZeroPadding(-2, -2, -2, -2)) -- identity + de-padding
       seq:add(con)