plot eval alignment score right

This commit is contained in:
erogol
2021-01-06 13:19:40 +01:00
parent 3fa408a5ea
commit 8971c59b2d
+1 -1
View File
@@ -317,7 +317,7 @@ def evaluate(data_loader, model, criterion, ap, global_step, epoch):
epoch_time += step_time
# compute alignment score
align_error = 1 - alignment_diagonal_score(alignments)
align_error = 1 - alignment_diagonal_score(alignments, binary=True)
loss_dict['align_error'] = align_error
# aggregate losses from processes