Update loss.py (WongKinYiu#1337)

Update line 687: add a round bracket which make an error when start training
mxm1994 · Dec 29, 2022 · 48052c4 · 48052c4
1 parent 45f2c1f
commit 48052c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/loss.py b/utils/loss.py
@@ -682,7 +682,7 @@ def build_targets(self, p, targets, imgs):
                 all_gj.append(gj)
                 all_gi.append(gi)
                 all_anch.append(anch[i][idx])
-                from_which_layer.append((torch.ones(size=(len(b),)) * i).to(device)
+                from_which_layer.append((torch.ones(size=(len(b),)) * i).to(device))
 
                 fg_pred = pi[b, a, gj, gi]                
                 p_obj.append(fg_pred[:, 4:5])