Jelajahi Sumber

Fix shortcut connection order in unet_branch

nagadomi 6 tahun lalu
induk
melakukan
3654ee41cf
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      lib/srcnn.lua

+ 1 - 1
lib/srcnn.lua

@@ -568,8 +568,8 @@ local function unet_branch(backend, insert, backend, n_input, n_output, depad)
    block:add(insert)
    block:add(insert)
    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))
-   con:add(nn.SpatialZeroPadding(-depad, -depad, -depad, -depad))
    con:add(block)
    con:add(block)
+   con:add(nn.SpatialZeroPadding(-depad, -depad, -depad, -depad))
    model:add(con)
    model:add(con)
    model:add(nn.CAddTable())
    model:add(nn.CAddTable())
    return model
    return model