From c5540d80cc4e8918a93833d389c2a83ddf98afca Mon Sep 17 00:00:00 2001 From: erogol Date: Thu, 12 Mar 2020 03:27:24 +0100 Subject: [PATCH] bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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