From 3d1db0b5acee444085aeec488c15f543ca26ca79 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 19 Jan 2020 11:37:12 -0800 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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