‹ projects

cluster-rnn

a distributed Torch7 RNN cluster over MPI
Log | Files | Refs | README

commit a54b28c638de5c4b6dfd671ee27b64e3bbcef7ce
parent 589bba0e4e37c9b83cd7242b377cd5955f0f8114
Author: umhau <umhau@users.noreply.github.com>
Date:   Wed, 15 Feb 2017 14:27:47 -0500

shorten training time

Diffstat:
Mtrain.lua | 2+-
1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.lua b/train.lua @@ -50,7 +50,7 @@ cmd:option('-dropout',0,'dropout for regularization, used after each RNN hidden cmd:option('-recurrent_dropout',0,'dropout for regularization, used on recurrent connections. 0 = no dropout') cmd:option('-seq_length',50,'number of timesteps to unroll for') cmd:option('-batch_size',50,'number of sequences to train on in parallel') -cmd:option('-max_epochs',2,'number of full passes through the training data') +cmd:option('-max_epochs',0.5,'number of full passes through the training data') cmd:option('-grad_clip',5,'clip gradients at this value') cmd:option('-train_frac',0.95,'fraction of data that goes into train set') cmd:option('-val_frac',0.05,'fraction of data that goes into validation set')