From 6fb14fc903fcea68239541a3de0fca5e6dc036e7 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 5 Mar 2019 17:14:40 +0100
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index e12a0937..3fd6972f 100644
--- a/train.py
+++ b/train.py
@@ -87,7 +87,7 @@ def train(
 
     t0 = time.time()
     model_info(model)
-    n_burnin = min(dataloader.nB / 5, 1000)  # number of burn-in batches
+    n_burnin = min(round(dataloader.nB / 5), 1000)  # number of burn-in batches
     for epoch in range(epochs):
         epoch += start_epoch