关于batch size/epoch/dataloader

通常 dataloader 中可设置 batch size，dataloader会将图像进行一个打包成tensor加快训练。
假设训练集有8张图像，batch_size=4,epoch=2,那么每次训练，dataloader迭代器将会送1个tensor（4张图像）的数据，送2次即完成了整个训练集的遍历，即1个epoch。