Generalized regression criterion renaming (#1120)
This commit is contained in:
1 parent
10c85bf4eb
commit
0ada058f63
6 files changed
+22
-22
No files matched your search
@@ -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()
|
||||
|
||||
Reference in new issue
Block a user