进度条设置及信息打印(tqdm-python)
有时候自己写模型训练过程中,想要有点逼格的打印输出显示,可参考本文的方法,本文使用tqdm显示进度条,可实现前面与后面添加内容。
总结:
语言:python
库:tqdm
结果:进度条显示,进度条前后可添加信息。
代码如下:
if __name__ == '__main__': train_loader=100000 with tqdm(total=train_loader) as pbar: for iter, data in enumerate(range(train_loader)): ''' 循环体内内容 ''' #################### 打印信息控制############ pbar.set_description("epoch {}|{}".format(200, iter)) # 进度条前加内容 pbar.set_postfix(iter_all='{}||{}'.format(iter*300, iter), loss=iter/100000000) pbar.update()
代码显示如下:
模型抒写显示如下: