From 5afd90c900f0811bd395f7dd58557d8b022abda6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 12 Oct 2019 11:22:50 +0200 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)