使用tqdm组件构造程序进度条

使用tqdm组件构造程序进度条

觉得有用的话,欢迎一起讨论相互学习~

主要代码

import tqdm  # 引用tqdm组件
TRAIN_STEPS = N
for i in tqdm.tqdm(range(TRAIN_STEPS)): #用tqdm结构包含原有迭代器

效果演示

• 此处设置N=3000，为一个基本的MNIST手写数字识别程序