tqdm

ultralytics · Oct 12, 2020 · 4cebfc8 · 4cebfc8
1 parent 27ddecd
commit 4cebfc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'