Generalized regression criterion renaming (#1120)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-10-11 17:25:17 +02:00
1 parent 10c85bf4eb
commit 0ada058f63
6 files changed
+22 -22

No files matched your search

+1 -1
View File
@@ -106,7 +106,7 @@ def test(data,
# Compute loss
if training: # if model has loss hyperparameters
loss += compute_loss([x.float() for x in train_out], targets, model)[1][:3] # GIoU, obj, cls
loss += compute_loss([x.float() for x in train_out], targets, model)[1][:3] # box, obj, cls
# Run NMS
t = time_synchronized()