在 PyTorch 中使用 tqdm
显示训练进度条,并解决常见错误TypeError: 'module' object is not callable
在进行深度学习模型训练时,尤其是在处理大规模数据时,实时了解训练过程中的进展是非常重要的。为了实现这一点,我们可以使用 tqdm
库,它可以非常方便地为你提供进度条显示。
1. 什么是 tqdm
?
TQDM
是一个快速、可扩展的 Python 进度条库。它可以用来显示迭代的进度,帮助我们实时了解程序运行的状态。tqdm
可以用于任何可迭代对象,如列表、train_loader
等。
安装 tqdm
如果你还没有安装 tqdm
,可以通过以下命令安装:
pip install tqdm
2. 如何使用 tqdm
包装 train_loader
?
在训练过程中,我们通常会使用 for
循环迭代数据加载器 (train_loader
) 来训练模型。通过使用 tqdm
包装这个迭代器,我们可以在训练时实时显示进度条。
正确的使用方法
from tqdm import tqdm # 导入 tqdm
# 假设你已经定义了 train_loader
for epoch in range(num_epochs):
model.train() # 设置模型为训练模式
running_loss = 0.0
correct = 0
total =