Fix shortcut connection order in unet_branch
This commit is contained in:
parent
2e59d97b17
commit
3654ee41cf
|
@ -568,8 +568,8 @@ local function unet_branch(backend, insert, backend, n_input, n_output, depad)
|
|||
block:add(insert)
|
||||
block:add(SpatialFullConvolution(backend, n_output, n_output, 2, 2, 2, 2, 0, 0))-- upsampling
|
||||
block:add(nn.LeakyReLU(0.1, true))
|
||||
con:add(nn.SpatialZeroPadding(-depad, -depad, -depad, -depad))
|
||||
con:add(block)
|
||||
con:add(nn.SpatialZeroPadding(-depad, -depad, -depad, -depad))
|
||||
model:add(con)
|
||||
model:add(nn.CAddTable())
|
||||
return model
|
||||
|
|
Loading…
Reference in a new issue