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'