minor rearrange

pkvk · Aug 23, 2017 · 6d4c8da · 6d4c8da
1 parent e684f0a
commit 6d4c8da
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/models/DenseConnectLayer.lua b/models/DenseConnectLayer.lua
@@ -27,7 +27,10 @@ function DenseConnectLayerStandard(nChannels, opt)
    net:add(cudnn.SpatialConvolution(nChannels, opt.growthRate, 3, 3, 1, 1, 1, 1))
    if opt.dropRate > 0 then net:add(nn.Dropout(opt.dropRate)) end
 
-   return net
+   return nn.Sequential()
+      :add(nn.Concat(2)
+         :add(nn.Identity())
+         :add(net))  
 end
 
 --------------------------------------------------------------------------------

diff --git a/models/densenet.lua b/models/densenet.lua
@@ -29,9 +29,7 @@ local function createModel(opt)
       if opt.optMemory >= 3 then
          model:add(nn.DenseConnectLayerCustom(nChannels, opt))
       else
-         model:add(nn.Concat(2)
-            :add(nn.Identity())
-            :add(DenseConnectLayerStandard(nChannels, opt)))      
+         model:add(DenseConnectLayerStandard(nChannels, opt))     
       end
    end