cluster-rnn

commit b5047961995ec865134708e5b29f3ecb2d241a80
parent 9532965f495522d60fef6243e80c3ae7e22ea625
Author: umhau <umhau@users.noreply.github.com>
Date:   Wed, 15 Feb 2017 14:18:35 -0500

removed optimm_state from checkpoint.  this allows saves, but prevents me from picking up where I left off.  Necessary, because the pclient settings don't like being saved.

Diffstat:
M
train.lua
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.lua b/train.lua
@@ -350,7 +350,7 @@ elseif opt.optimizer == 'eamsgd' then
     optimizer = optim.eamsgd
     optim_state.learningRate = mpiOptions.learningRate
     optim_state.momentum = mpiOptions.momentum
-    -- optim_state.pclient = mpiOptions.pclient
+    optim_state.pclient = mpiOptions.pclient
     optim_state.communicationPeriod = mpiOptions.communicationPeriod
     optim_state.movingRateAlpha = mpiOptions.movingRateAlpha
     optim_state.learningRateDecay = mpiOptions.learningRateDecay