From 5263608d12a059ee59c6598b8de8249309663285 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 26 Apr 2019 14:49:40 +0200
Subject: [PATCH] updates

---
 train.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 98d69538..f30ba066 100644
--- a/train.py
+++ b/train.py
@@ -143,7 +143,7 @@ def train(
         model, optimizer = amp.initialize(model, optimizer, opt_level='O1')
 
     # Start training
-    t = time.time()
+    t, t0 = time.time(), time.time()
     model.hyp = hyp  # attach hyperparameters to model
     model_info(model)
     nb = len(dataloader)
@@ -255,6 +255,8 @@ def train(
             # Delete checkpoint
             del chkpt
 
+    dt = (time.time() - t0) / 3600
+    print('%g epochs completed in %.3f hours.' % (epoch - start_epoch, dt))
     return results