diff --git a/train.py b/train.py index b3a0589b..d2509c72 100644 --- a/train.py +++ b/train.py @@ -409,7 +409,7 @@ def evaluate(model, criterion, ap, global_step, epoch): loss_dict['ga_loss'].item(), keep_avg['avg_ga_loss'], align_score, keep_avg['avg_align_score']), - flush=Tr ue) + flush=True) if args.rank == 0: # Diagnostic visualizations