cluster-rnn

commit 71251f886db6cb5bd4d1e4cde1685fa359565708
parent ba4e68567caa4e13fa2c15a99993579bcadb91a0
Author: umhau <umhau@users.noreply.github.com>
Date:   Wed, 15 Feb 2017 17:59:05 -0500

allow fractional epochs

Diffstat:
M
train.lua
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.lua b/train.lua
@@ -332,7 +332,7 @@ end
 -- start optimization here
 train_losses = {}
 val_losses = {}
-local iterations = opt.max_epochs * loader.ntrain
+local iterations = math.floor(opt.max_epochs * loader.ntrain)
 local iterations_per_epoch = loader.ntrain
 local loss0 = nil