From 4cebfc8c44fa0771c062bcec57e03331f8838bc1 Mon Sep 17 00:00:00 2001
From: Jirka <jirka.borovec@seznam.cz>
Date: Tue, 29 Sep 2020 09:08:50 +0200
Subject: [PATCH] tqdm

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

diff --git a/train.py b/train.py
index 8bb5dfda1ab8..563916529ace 100644
--- a/train.py
+++ b/train.py
@@ -495,7 +495,7 @@ def train(hyp, opt, device, tb_writer=None):
         if opt.bucket:
             os.system('gsutil cp gs://%s/evolve.txt .' % opt.bucket)  # download evolve.txt if exists
 
-        for _ in range(300):  # generations to evolve
+        for _ in tqdm(range(300), desc='perform evolve >>'):  # generations to evolve
             if os.path.exists('evolve.txt'):  # if evolve.txt exists: select best hyps and mutate
                 # Select parent(s)
                 parent = 'single'  # parent selection method: 'single' or 'weighted'