diff --git a/models/openface/resnet1.def.lua b/models/openface/resnet1.def.lua
index 90b430b..0461d21 100644
--- 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))