Update loss.py (#1337)

Update line 687: add a round bracket which make an error when start training
2022-12-29 22:45:30 +07:00 · 2022-12-29 22:45:30 +07:00 · 48052c42c4
parent 45f2c1fe55
commit 48052c42c4
1 changed files with 1 additions and 1 deletions
--- a/utils/loss.py
+++ b/utils/loss.py
@ -682,7 +682,7 @@ class ComputeLossOTA:
                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])