Fix srresnet_2x
This commit is contained in:
parent
bdaca16c67
commit
4be1141292
|
@ -403,6 +403,7 @@ function srcnn.srresnet_2x(backend, ch)
|
||||||
conv:add(SpatialConvolution(backend, 64, 64, 3, 3, 1, 1, 0, 0))
|
conv:add(SpatialConvolution(backend, 64, 64, 3, 3, 1, 1, 0, 0))
|
||||||
conv:add(ReLU(backend))
|
conv:add(ReLU(backend))
|
||||||
conv:add(SpatialConvolution(backend, 64, 64, 3, 3, 1, 1, 0, 0))
|
conv:add(SpatialConvolution(backend, 64, 64, 3, 3, 1, 1, 0, 0))
|
||||||
|
conv:add(ReLU(backend))
|
||||||
con:add(conv)
|
con:add(conv)
|
||||||
con:add(nn.SpatialZeroPadding(-2, -2, -2, -2)) -- identity + de-padding
|
con:add(nn.SpatialZeroPadding(-2, -2, -2, -2)) -- identity + de-padding
|
||||||
seq:add(con)
|
seq:add(con)
|
||||||
|
|
Loading…
Reference in a new issue