Update TQDM bar format (#6988)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-03-15 16:32:56 +01:00
1 parent 932dc78496
commit c09fb2aa95
2 files changed
+5 -4

No files matched your search

+1 -1
View File
@@ -152,7 +152,7 @@ def kmean_anchors(dataset='./data/coco128.yaml', n=9, img_size=640, thr=4.0, gen
# Evolve
f, sh, mp, s = anchor_fitness(k), k.shape, 0.9, 0.1 # fitness, generations, mutation prob, sigma
pbar = tqdm(range(gen), desc=f'{PREFIX}Evolving anchors with Genetic Algorithm:') # progress bar
pbar = tqdm(range(gen), bar_format='{l_bar}{bar:10}{r_bar}{bar:-10b}') # progress bar
for _ in pbar:
v = np.ones(sh)
while (v == 1).all(): # mutate until a change occurs (prevent duplicates)