可以创造一个可视化的进度条,大概是这个效果:
pip install tqdm用法
最简单的举例:
from tqdm import tqdm
import time
k=0
for i in tqdm(range(1000)):
k=k+i
time.sleep(0.01)
效果:
这个进度条
改进下:
k=0
for i in tqdm(range(10)):
k=k+i
time.sleep(1)
print("hi")
每当循环到print函数时,他就会打印每一次的进度:
如果是在模型训练过程中,可以这样:
from tqdm import tqdm
for epoch in tqdm(range(EPOCHS), desc="Epoch", unit="epoch"):
train(net, data_loader, optimizer, epoch + 1) # train 函数 具体内容此处省略
效果:



