eg1:
test数据集 1000张图片
可以看到训练过程中,程序的具体运行轨迹。
eg2:
训练数据集4000张,batch_size 是 100,*号后input() 只需要键入一次,即完成20轮次的训练。
循环是在
for _ in 被加载的数据集 :
{ }
这个部分重复执行的。
len(trainloader)是 4000/100 =40 个,
也就是说 加载后的数据集 包含了40个部分,每一部分作为[100,3,224,224]的矩阵只执行一次,共执行40次。
外面的
for _ in range(epoch):
{ }
这个部分被循环执行 epoch 次,此处是20次,换句话说,对4000张图片执行了20次训练。