Add commented Avg to resnet definition.

2016-06-19 17:14:07 -04:00 · 2016-06-19 17:14:07 -04:00 · a24484c671
parent 0574a72641
commit a24484c671
1 changed files with 2 additions and 1 deletions
--- a/models/openface/resnet1.def.lua
+++ b/models/openface/resnet1.def.lua
@ -139,7 +139,8 @@ function createModel()
   model:add(layer(block, 128, def[2], 2))
   model:add(layer(block, 256, def[3], 2))
   model:add(layer(block, 512, def[4], 2))
-   -- model:add(nn.Reshape(nLinear))
+   -- TODO: Add back?
+   -- model:add(Avg(7, 7, 1, 1))
   model:add(nn.View(nLinear))
   model:add(nn.Linear(nLinear, opt.embSize))
   model:add(nn.Normalize(2))