pieces of cake concerning torchtorchvision

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()