Fix bug in trainer where not noting all tasks' training progress
@@ -856,6 +856,7 @@ class SamplingMultiTaskTrainer:
# Validation
if n_pass % (validation_interval) == 0:
# Get metrics for all training progress so far
+ for task in tasks:
task_metrics = task.get_metrics(reset=True)
for name, value in task_metrics.items():
all_tr_metrics["%s_%s" % (task.name, name)] = value
