bug fix

2020-03-12 03:27:24 +01:00 · 2020-03-12 03:27:24 +01:00 · c5540d80cc
parent cbcdec83da
commit c5540d80cc
1 changed files with 1 additions and 1 deletions
--- 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