cluster-rnn

commit b51fc88f00a08f3fcc03c2a36e6208538b24fce3
parent 8455458833d9633e1d703835fbfc90c8dd569485
Author: umhau <umhau@users.noreply.github.com>
Date:   Tue, 14 Feb 2017 21:30:30 -0500

shorten training time for testing

Diffstat:
M
train.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',50,'number of full passes through the training data')
+cmd:option('-max_epochs',2,'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')