updates
This commit is contained in:
parent
85fbb903f7
commit
3d1db0b5ac
2
train.py
2
train.py
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue