1. version match
torchvision的版本对应关系
2. utilize tqdm to present process bar
lay a pbar
from tqdm import tqdm
pbar = tqdm(unit="batch", file=sys.stdout,total=len(self.training_dataloader)) #处理单位为batch
pbar2 = tqdm(range(20), desc='It\'s a test')
iterable=None
,可迭代对象。如上一节中的range(20)
desc=None
,传入str类型,作为进度条标题。如上一节中的desc=‘It’s a test’
total=None
,预期的迭代次数。一般不填,默认为iterable的长度。
leave=True
,迭代结束时,是否保留最终的进度条。默认保留。
file=None
,输出指向位置,默认是终端,一般不需要设置。
ncols=None
,可以自定义进度条的总长度
unit
,描述处理项目的文字,默认’it’,即100it/s;处理照片设置为’img’,则为100img/s
postfix
,以字典形式传入详细信息,将显示在进度条中。例如postfix={‘value’: 520}
unit_scale
,自动根据国际标准进行项目处理速度单位的换算,例如100000it/s换算为100kit/s
set postfix
pbar.set_postfix({"loss": np.mean(avg_loss), "acc": np.mean(avg_accuracy) * 100.0}
set update frequency
pbar.update(1)
pbar.close()