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

Remove deprecated --arc architecture options, implement --arc default for all cases

Browse Source
This commit is contained in:
Glenn Jocher
2020-03-16 20:46:25 -07:00
parent 77c6c01970
commit 448c4a6e1f
3 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/utils.py
+3 -3
View File
@@ -377,7 +377,6 @@ def compute_loss(p, targets, model): # predictions, targets, model
lcls, lbox, lobj = ft([0]), ft([0]), ft([0])
tcls, tbox, indices, anchor_vec = build_targets(model, targets)
h = model.hyp # hyperparameters
arc = model.arc # architecture
red = 'mean' # Loss reduction (sum or mean)
# Define criteria
@@ -388,8 +387,9 @@ def compute_loss(p, targets, model): # predictions, targets, model
cp, cn = smooth_BCE(eps=0.0)
# focal loss
if 'F' in arc:
BCEcls, BCEobj = FocalLoss(BCEcls, h['fl_gamma']), FocalLoss(BCEobj, h['fl_gamma'])
g = h['fl_gamma'] # focal loss gamma
if g > 0:
BCEcls, BCEobj = FocalLoss(BCEcls, g), FocalLoss(BCEobj, g)
# Compute losses
np, ng = 0, 0 # number grid points, targets