This commit is contained in:
erogol 2020-03-12 03:27:24 +01:00
parent cbcdec83da
commit c5540d80cc
1 changed files with 1 additions and 1 deletions

View File

@ -409,7 +409,7 @@ def evaluate(model, criterion, ap, global_step, epoch):
loss_dict['ga_loss'].item(), loss_dict['ga_loss'].item(),
keep_avg['avg_ga_loss'], keep_avg['avg_ga_loss'],
align_score, keep_avg['avg_align_score']), align_score, keep_avg['avg_align_score']),
flush=Tr ue) flush=True)
if args.rank == 0: if args.rank == 0:
# Diagnostic visualizations # Diagnostic visualizations