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

Fix off-by-one error in schedules

parent ff8e4281

ahisto_named_entity_search/recognition/schedule.py

+2 −2
Original line number Diff line number Diff line
@@ -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