Temporarily remove optimize_net for #127.

cmusatyalab · Jun 7, 2016 · 21f1bcc · 21f1bcc
1 parent 40f124c
commit 21f1bcc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/training/model.lua b/training/model.lua
@@ -38,7 +38,7 @@ if opt.cuda then
    criterion:cuda()
 end
 
-optimizeNet(model, opt.imgDim)
+-- optimizeNet(model, opt.imgDim)
 
 print('=> Model')
 print(model)

diff --git a/training/train.lua b/training/train.lua
@@ -22,7 +22,6 @@ require 'torchx' --for concetration the table of tensors
 
 paths.dofile("OpenFaceOptim.lua")
 
-
 local optimMethod = optim.adam
 local optimState = {} -- Use for other algorithms like SGD
 local optimator = OpenFaceOptim(model, optimState)
@@ -88,9 +87,9 @@ function train()
 
    collectgarbage()
 
-   local nnModel = model:float():clone():clearState()
+   local nnModel = model:float():clearState()
    if opt.cudnn then
-    cudnn.convert(nnModel,nn)
+      cudnn.convert(nnModel, nn)
    end
 
    torch.save(paths.concat(opt.save, 'model_' .. epoch .. '.t7'), nnModel)