引入该模块:
from tqdm import tqdm
tqdm可用于所有可迭代对象,所以可以直接用在dataloader上,如下:
for data, target in tqdm(train_loader):
...
效果:
tqdm用在dataloader上其实是对每个batch和batch总数做的进度条,我也是在训练比较慢时写出来看看训练到哪了。。。
引入该模块:
from tqdm import tqdm
tqdm可用于所有可迭代对象,所以可以直接用在dataloader上,如下:
for data, target in tqdm(train_loader):
...
效果:
tqdm用在dataloader上其实是对每个batch和batch总数做的进度条,我也是在训练比较慢时写出来看看训练到哪了。。。