nagadomi 6 년 전
부모
커밋
8051535fdc
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      lib/srcnn.lua

+ 2 - 2
lib/srcnn.lua

@@ -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