0
点赞
收藏
分享

微信扫一扫

for XX in dataloader和for XX in enumerate(dataloader)的区别

孟祥忠诗歌 2022-02-25 阅读 123

通过用MNIST数据集和CNN网络模型做实验得知:

 for i, inputs in train_loader:

不加enumerate的话只能返回两个值,其中第一个值(这里是i)为输入的图片数据,第二个值为数据标签;

for i, (inputs,labels) in enumerate(train_loader): 

加上enumerate,可以返回三个值,第一个值为序号,第二个值是输入数据,第三个值是数据标签。

举报

相关推荐

0 条评论