通过用MNIST数据集和CNN网络模型做实验得知:
for i, inputs in train_loader:
不加enumerate的话只能返回两个值,其中第一个值(这里是i)为输入的图片数据,第二个值为数据标签;
for i, (inputs,labels) in enumerate(train_loader):
加上enumerate,可以返回三个值,第一个值为序号,第二个值是输入数据,第三个值是数据标签。
通过用MNIST数据集和CNN网络模型做实验得知:
for i, inputs in train_loader:
不加enumerate的话只能返回两个值,其中第一个值(这里是i)为输入的图片数据,第二个值为数据标签;
for i, (inputs,labels) in enumerate(train_loader):
加上enumerate,可以返回三个值,第一个值为序号,第二个值是输入数据,第三个值是数据标签。