updates

2020-01-19 11:37:12 -08:00 · 2020-01-19 11:37:12 -08:00 · 3d1db0b5ac
parent 85fbb903f7
commit 3d1db0b5ac
1 changed files with 1 additions and 1 deletions
--- 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