OrgLion-ML
/
openface
mirror of https://github.com/cmusatyalab/openface.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Add commented Avg to resnet definition.

This commit is contained in:
Brandon Amos 2016-06-19 17:14:07 -04:00
parent 0574a72641
commit a24484c671
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/openface/resnet1.def.lua
View File

@ -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))