cluster-rnn

commit cfcf0469a981484eb6c5cbb55576e350908f5728
parent 1c464b482a0eb06614aacf33716f42ae6805bf65
Author: umhau <umhau@users.noreply.github.com>
Date:   Wed, 15 Feb 2017 14:08:11 -0500

pclient data should not be part of the optim_state variable

Diffstat:
M
train.lua
 | 
4
++--

1 file changed, 2 insertions(+), 2 deletions(-)
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
@@ -361,7 +361,7 @@ end
 
 -- initialize MPI optimizer clients
 rank = mpiOptions.rank or -1
-pclient = optim_state.pclient or nil
+pclient = mpiOptions.pclient or nil
 print('i am ' .. rank .. ' ready to run')
 if pclient then
    pclient:start(params,grad_params)