Training: Use ADAM instead of adadelta.

cmusatyalab · Jun 3, 2016 · 130edce · 130edce
1 parent 02b85b7
commit 130edce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/training/train.lua b/training/train.lua
@@ -23,7 +23,7 @@ require 'torchx' --for concetration the table of tensors
 paths.dofile("OpenFaceOptim.lua")
 
 
-local optimMethod = optim.adadelta
+local optimMethod = optim.adam
 local optimState = {} -- Use for other algorithms like SGD
 local optimator = OpenFaceOptim(model, optimState)
 
@@ -87,7 +87,7 @@ function train()
    print('\n')
 
    collectgarbage()
-   
+
    local nnModel = model:float():clone():clearState()
    if opt.cudnn then
     cudnn.convert(nnModel,nn)