【tqdm】进度条工具

学渣渣渣渣渣

已于 2022-10-26 12:29:14 修改

阅读量1.8w

点赞数

CC 4.0 BY-SA版权

分类专栏：方法 pytorch 文章标签： python 深度学习人工智能

于 2022-10-09 13:33:51 首次发布

本文链接：https://blog.csdn.net/weixin_42468475/article/details/127224625

pytorch 同时被 2 个专栏收录

34 篇文章

订阅专栏

方法

31 篇文章

订阅专栏

本文介绍tqdm库的基本用法，通过示例展示如何在循环中使用tqdm来显示进度条，并演示不同场景下进度条的变化情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tqdm 是一种呈现循环进度的工具包，本文讲讲述他的常用用法。

Example 1

假设FL框架中一共有10个用户，每次随机抽取5个用户进行更新，一共训练6个epochs，结合tqdm显示训练进程的代码框架如下：

for epoch in range(6):
    print("epoch:", epoch)
    for idx in tqdm(np.random.choice(range(10), 5, replace=False), desc='Epoch:%d, lr:%f' % (100, 0.5)):
        print(idx)
        #time.sleep(2)

out：
在这里插入图片描述

如果说我添加一行 time.sleep(2)，那效果就不一样了：

for epoch in range(6):
    print("epoch:", epoch)
    for idx in tqdm(np.random.choice(range(10), 5, replace=False), desc='Epoch:%d, lr:%f' % (100, 0.5)):
        print(idx)
        time.sleep(2)