tqdm双进度条训练模型

zibingoo

已于 2024-03-22 08:17:20 修改

阅读量497

点赞数 7

文章标签：深度学习人工智能

于 2024-02-02 13:57:26 首次发布

本文链接：https://blog.csdn.net/weixin_46079657/article/details/135992431

版权

效果图

例子一

from tqdm import tqdm
import time
pbar1 = tqdm(total=5, position=0, desc="Epoch", leave=False, colour='green', ncols=100,unit="epoch")
for i in range(5):
    pbar2 = tqdm(total=10, position=1, desc="Iteration", leave=False, colour='red', ncols=100,unit="iteration")
    for j in range(10):
        time.sleep(0.2)
        pbar2.update(1)
    pbar1.update(1)

例子二

for i in tqdm(range(5),total=5, position=0, desc="Epoch", leave=False, colour='green', ncols=100,unit="epoch"):
    for j in tqdm(range(10),total=10, position=1, desc="Iteration", leave=False, colour='red', ncols=100,unit="iteration"):
        time.sleep(0.2)

例子三

for epoch in range(num_epochs):
	for i,(inputs, labels) in enumerate(train_loader,0):

可以改成以下方式

for epoch in tqdm(range(num_epochs),total=num_epochs, position=0, desc="Epoch", leave=False, colour='green', ncols=100,unit="epoch"):
	for i,(inputs, labels) in enumerate(tqdm(train_loader,total=len(train_loader), position=1, desc="Iteration", leave=False, colour='red', ncols=100,unit="iteration")):