cluster-rnn

commit 9fea467f01172e577c86e786b297546d6d71bca6
parent 9640ddad4b2981a55d64c56e83efafb9511b7e98
Author: umhau <umhau@users.noreply.github.com>
Date:   Wed, 15 Feb 2017 00:56:05 -0500

make more mpi parameters available

Diffstat:
M
train.lua
 | 
2
++

1 file changed, 2 insertions(+), 0 deletions(-)
diff --git a/train.lua b/train.lua
@@ -353,6 +353,8 @@ elseif opt.optimizer == 'eamsgd' then
     optim_state.pclient = mpiOptions.pclient
     optim_state.communicationPeriod = mpiOptions.communicationPeriod
     optim_state.movingRateAlpha = mpiOptions.movingRateAlpha
+    optim_state.learningRateDecay = mpiOptions.learningRateDecay
+    optim_state.learningRateDecayPower = mpiOptions.learningRateDecayPower
 else
     optimizer = optim.rmsprop
 end