|
@@ -568,8 +568,8 @@ local function unet_branch(backend, insert, backend, n_input, n_output, depad)
|
|
block:add(SpatialFullConvolution(backend, n_output, n_output, 2, 2, 2, 2, 0, 0))-- upsampling
|
|
block:add(SpatialFullConvolution(backend, n_output, n_output, 2, 2, 2, 2, 0, 0))-- upsampling
|
|
block:add(nn.LeakyReLU(0.1, true))
|
|
block:add(nn.LeakyReLU(0.1, true))
|