nn4.small2.def: Comment out reshapes.

2016-02-06 18:56:50 -05:00 · 2016-02-06 18:56:50 -05:00 · b43f510489
parent c206fa20c5
commit b43f510489
1 changed files with 2 additions and 2 deletions
--- a/models/openface/nn4.small2.def.lua
+++ b/models/openface/nn4.small2.def.lua
@ -122,7 +122,7 @@ function createModel()
              pool = nn.SpatialLPPooling(960, 2, 3, 3),
              batchNorm = true
   })
-   net:add(nn.Reshape(736,3,3))
+   -- net:add(nn.Reshape(736,3,3))

   -- Inception (5b)
   net:add(nn.Inception{
@ -138,7 +138,7 @@ function createModel()
   net:add(nn.SpatialAveragePooling(3, 3))

   -- Validate shape with:
-   net:add(nn.Reshape(736))
+   -- net:add(nn.Reshape(736))

   net:add(nn.View(736))
   net:add(nn.Linear(736, opt.embSize))