参考:https://blog.csdn.net/winter2121/article/details/111356587
总结:tqdm()只是给遍历过程添加了进度条,传入的可以是可迭代对象,也可以是不可迭代对象(需要额外设置,成为手动更新)。这两种运行模式具体见下面链接:
https://zhuanlan.zhihu.com/p/163613814
# 可迭代对象,自动更新
dic = ['a', 'b', 'c', 'd', 'e']
pbar = tqdm(dic)
for i in pbar:
pbar.set_description('Processing '+i)
time.sleep(0.2)
Processing e: 100%|██████████| 5/5 [00:01<00:00, 4.69it/s]
# 不可迭代对象,手动更新
from tqdm import tqdm
import time
with tqdm(range(100), desc='Test') as tbar:
for i in tbar:
tbar.set_postfix(loss=i/100, x=i)
tbar.update() # 手动更新,默认参数n=1,每update一次,进度+n
time.sleep(0.2)
# 不可迭代对象,手动更新
from tqdm import tqdm
import time
with tqdm(range(100), desc='Test') as tbar:
for i in tbar:
tbar.set_postfix({'loss': '{0:1.5f}'.format(i/100), "x": '{0:1.5f}'.format(i)})
tbar.update() # 手动更新,默认参数n=1,每update一次,进度+n
time.sleep(0.2)
# 不可迭代对象,手动更新
from tqdm import tqdm
import time
total = 200 #总迭代次数
loss = total
with tqdm(total=200, desc='Test') as pbar:
pbar.set_description('Processing:') # tqdm()中的desc参数和pbar.set_description()是一回事,后者优先。
for i in range(20):
loss -= 1
pbar.set_postfix({'loss': '{0:1.5f}'.format(i/100), "x": '{0:1.5f}'.format(i)}) # 输入一个字典,显示实验指标
# pbar.set_postfix(loss=i/100, x=i)
pbar.update(10) # 手动更新,默认参数n=1,每update一次,进度+n
time.sleep(0.2)
总结:tqdm.set_postfix()传入参数的两种方式:
# 1 传入元组
tbar.set_postfix(loss=i/100, x=i)
# 2 传入字典
tbar.set_postfix({'loss': '{0:1.5f}'.format(i/100), "x": '{0:1.5f}'.format(i)})
# 两种方法的结果一样