From 8b6c8a53182b2415fd61459fc9a0ccbdef8dc904 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 4 Mar 2020 16:33:14 -0800
Subject: [PATCH] updates

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index ef3faca2..5ca0efe4 100644
--- a/train.py
+++ b/train.py
@@ -215,7 +215,7 @@ def train():
 
         # Prebias
         if prebias:
-            ne = 3  # number of prebias epochs
+            ne = max(round(30 / nb), 3)  # number of prebias epochs
             ps = np.interp(epoch, [0, ne], [0.1, hyp['lr0'] * 2]), \
                  np.interp(epoch, [0, ne], [0.9, hyp['momentum']])  # prebias settings (lr=0.1, momentum=0.9)
             if epoch == ne: