apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update loss.py (#1337)

Browse Source 
Update line 687: add a round bracket which make an error when start training
This commit is contained in:
Trieu Viet Hung 2022-12-29 22:45:30 +07:00 committed by GitHub
parent 45f2c1fe55
commit 48052c42c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/loss.py
View File

@ -682,7 +682,7 @@ class ComputeLossOTA:
all_gj.append(gj) all_gj.append(gj)
all_gi.append(gi) all_gi.append(gi)
all_anch.append(anch[i][idx]) 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] fg_pred = pi[b, a, gj, gi]
p_obj.append(fg_pred[:, 4:5]) p_obj.append(fg_pred[:, 4:5])