enumerate和iter的使用
enumerate和iter的使用
卷积神经网络中对数据集需要进行解析出图片信息和标签信息传输到model中,因此需要enumerate 和iter其中一种就行
一、enumerat的使用
for step, data in enumerate(train_dataloader):
images, labels = data
二、iter的使用
train_data_iter = iter(train_dataloader)#转换为迭代器
trian_image, train_label = tr
原创
2022-04-03 15:39:15 ·
1936 阅读 ·
0 评论