Commit 832adbaa authored by Vít Novotný's avatar Vít Novotný
Browse files

Prevent future objectives from affecting early stopping (cont.)

parent a3022d22
Pipeline #146498 passed with stage
in 8 minutes and 38 seconds
......@@ -46,7 +46,8 @@ class FineTuningSchedule(Schedule):
while True:
# Prevent future objectives from affecting early stopping
for remaining_objective in remaining_objectives:
remaining_objective.evaluations_history["eval"] = {}
for evaluator, values in remaining_objective.evaluations_history["eval"].items():
remaining_objective.evaluations_history["eval"][evaluator] = []
if objective in self.converged_objectives and not self.args.log_converged_objectives:
break
num_train_epochs = objective.epoch - starting_epoch
......
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