From 48052c42c4f913ed83dfa23e1469e833a880eaa6 Mon Sep 17 00:00:00 2001
From: Trieu Viet Hung <69461968+hungtrieu07@users.noreply.github.com>
Date: Thu, 29 Dec 2022 22:45:30 +0700
Subject: [PATCH] Update loss.py (#1337)

Update line 687: add a round bracket which make an error when start training
---
 utils/loss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/loss.py b/utils/loss.py
index 5fc73eaf..a5b72885 100644
--- 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])