Fix interpretation of --max-epoch

e9967cd3 · Myle Ott · 678d0bbc · e9967cd3
Commit e9967cd3 authored Jun 21, 2018 by Myle Ott
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

train.py train.py +1 -1

No files found.
--- a/train.py
+++ b/train.py
@@ -82,7 +82,7 @@ def main(args):
    train_meter.start()
    valid_losses = [None]
    valid_subsets = args.valid_subset.split(',')
-    while lr > args.min_lr and epoch_itr.epoch <= max_epoch and trainer.get_num_updates() < max_update:
+    while lr > args.min_lr and epoch_itr.epoch < max_epoch and trainer.get_num_updates() < max_update:
        # train for one epoch
        train(args, trainer, task, epoch_itr)