plot eval alignment score right

This commit is contained in:
erogol committed 2021-01-06 13:19:40 +01:00
1 parent 3fa408a5ea
commit 8971c59b2d
1 file changed
+1 -1
+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