back to rmsprop

coreylynch · Jan 31, 2016 · 342577a · 342577a
1 parent 1411398
commit 342577a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/train.lua b/train.lua
@@ -377,8 +377,7 @@ for i = 1, iterations do
     local epoch = i / loader.ntrain
 
     local timer = torch.Timer()
-    -- local _, loss = optim.rmsprop(feval, params, optim_state)
-    local _, loss = optim.adam(feval, params, optim_state)
+    local _, loss = optim.rmsprop(feval, params, optim_state)
     if opt.accurate_gpu_timing == 1 and opt.gpuid >= 0 then
         --[[
         Note on timing: The reported time can be off because the GPU is invoked async. If one