安装tqdm
$ conda install tqdm
$ #pip install tqdm
#导入tqdm包
from tqdm import tqdm
#在循环之前,定义tqdm
bar = tqdm(total=len(train_loader))
#在进度条之前显示内容设置(可选)
bar.set_description("Train:Epoch[%d/%d]" % (config.epoch + 1, num_epochs))
#进度条之后的显示内容设置(可选)
bar.set_postfix(Loss=loss.item())
#更新进度条
bar.update()
#关闭
bar.close()
实例
结果
网友评论