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

nn4.small2.def: Comment out reshapes.

This commit is contained in:
Brandon Amos 2016-02-06 18:56:50 -05:00
parent c206fa20c5
commit b43f510489
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models/openface/nn4.small2.def.lua
View File

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