From 4be1141292ae9c1be620843deb08621ea8029366 Mon Sep 17 00:00:00 2001 From: nagadomi Date: Sun, 6 Nov 2016 01:19:20 +0900 Subject: [PATCH] Fix srresnet_2x --- lib/srcnn.lua | 1 + 1 file changed, 1 insertion(+) diff --git a/lib/srcnn.lua b/lib/srcnn.lua index e7a49c8..afed09d 100644 --- a/lib/srcnn.lua +++ b/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)