Commit 837ffec0 authored by Vít Novotný's avatar Vít Novotný
Browse files

Fix off-by-one error in schedules

parent ff8e4281
Pipeline #146527 passed with stage
in 8 minutes and 59 seconds
......@@ -25,7 +25,7 @@ class FairSequentialSchedule(Schedule):
if objective in self.converged_objectives and not self.args.log_converged_objectives:
continue
num_train_epochs = objective.epoch - starting_epoch
if num_train_epochs >= self.MAX_NUM_TRAIN_EPOCHS:
if num_train_epochs > self.MAX_NUM_TRAIN_EPOCHS:
continue
yield objective
......@@ -51,7 +51,7 @@ class FineTuningSchedule(Schedule):
if objective in self.converged_objectives and not self.args.log_converged_objectives:
break
num_train_epochs = objective.epoch - starting_epoch
if num_train_epochs >= self.MAX_NUM_TRAIN_EPOCHS:
if num_train_epochs > self.MAX_NUM_TRAIN_EPOCHS:
break
yield objective
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment