From d777a57b9cb1e18d39b1beb8ce8967975cf29321 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 23 Aug 2019 13:39:43 +0200
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index efa71951..8b859539 100644
--- a/train.py
+++ b/train.py
@@ -236,7 +236,7 @@ def train():
         mloss = torch.zeros(5).to(device)  # mean losses
         pbar = tqdm(enumerate(dataloader), total=nb)  # progress bar
         for i, (imgs, targets, paths, _) in pbar:  # batch -------------------------------------------------------------
-            ni = (i + nb * epoch)  # number integrated batches (since train start)
+            ni = i + nb * epoch  # number integrated batches (since train start)
             imgs = imgs.to(device)
             targets = targets.to(device)