解决Python中tqdm库动态进度条不显示的问题

python的tqdm库不显示动态进度条的问题

本质原因是tqdm无法获取内部对象的长度,这可能是因为内部对象是一个迭代器,问题经常发生在同时使用tqdm与enumerate的场合,例如深度学习中经常可能出现的:

tqdm.tqdm(enumerate(train_loader))

解决方法也很简单,只需要调换顺序,让tqdm内部是一个可以获取长度的对象就可以了,可以修改为:

enumerate(tqdm.tqdm(train_loader))

作者:starvapour

物联沃分享整理
物联沃-IOTWORD物联网 » 解决Python中tqdm库动态进度条不显示的问题

发表回复