enumerate和iter的使用
卷积神经网络中对数据集需要进行解析出图片信息和标签信息传输到model中,因此需要enumerate 和iter其中一种就行
一、enumerate的使用
for step, data in enumerate(train_dataloader):
images, labels = data
二、iter的使用
train_data_iter = iter(train_dataloader)#转换为迭代器
trian_image, train_label = train_data_iter.next()#迭代器是指用next(it)函数获取可迭代对象的数据