diff --git a/utils/utils.py b/utils/utils.py index 627f8930..d7182103 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -356,7 +356,7 @@ def compute_loss(p, targets, model): # predictions, targets, model # with open('targets.txt', 'a') as file: # [file.write('%11.5g ' * 4 % tuple(x) + '\n') for x in torch.cat((txy[i], twh[i]), 1)] - if 'default' in arc: # seperate obj and cls + if 'default' in arc: # separate obj and cls lobj += BCEobj(pi[..., 4], tobj) # obj loss elif 'BCE' in arc: # unified BCE (80 classes)