From 16812495884a38fa6724aea5c136b6ee4ed93e26 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 14 Apr 2020 04:15:53 -0700 Subject: [PATCH] cleanup --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index def70e26..ef77e345 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -443,7 +443,7 @@ def build_targets(p, targets, model): anchor_vec = model.module.module_list[j].anchor_vec if multi_gpu else model.module_list[j].anchor_vec # iou of targets-anchors - gain[2:] = torch.tensor(p[j].shape)[[2, 3, 2, 3]] # xyxy gain + gain[2:] = torch.tensor(p[i].shape)[[2, 3, 2, 3]] # xyxy gain t, a = targets * gain, [] gwh = t[:, 4:6] if nt: