This commit is contained in:
Glenn Jocher 2020-01-19 11:37:12 -08:00
parent 85fbb903f7
commit 3d1db0b5ac
1 changed files with 1 additions and 1 deletions

View File

@ -221,7 +221,7 @@ def train():
# Prebias # Prebias
if prebias: if prebias:
if epoch < 1: # prebias if epoch < 3: # prebias
ps = 0.1, 0.9 # prebias settings (lr=0.1, momentum=0.9) ps = 0.1, 0.9 # prebias settings (lr=0.1, momentum=0.9)
else: # normal training else: # normal training
ps = hyp['lr0'], hyp['momentum'] # normal training settings ps = hyp['lr0'], hyp['momentum'] # normal training settings