fix fully connected layer in cnn_model

nagadomi · Aug 23, 2014 · 1cc9077 · 1cc9077
1 parent 2e7c0fa
commit 1cc9077
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cnn_model.lua b/cnn_model.lua
@@ -19,6 +19,7 @@ function cnn_model() -- validate.lua Acc: 0.88
 
    -- fully connected layers
    model:add(nn.SpatialConvolutionMM(512, 1024, 2, 2, 1, 1))
+   model:add(nn.ReLU())
    model:add(nn.Dropout(0.5))
    model:add(nn.SpatialConvolutionMM(1024, 10, 1, 1, 1, 1))