diff --git a/train.py b/train.py index 43a0465d..0db2f89b 100644 --- a/train.py +++ b/train.py @@ -221,7 +221,7 @@ def train(): # Prebias if prebias: - if epoch < 1: # prebias + if epoch < 3: # prebias ps = 0.1, 0.9 # prebias settings (lr=0.1, momentum=0.9) else: # normal training ps = hyp['lr0'], hyp['momentum'] # normal training settings