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 = train_data_iter.next()#迭代器是指用next(it)函数获取可迭代对象的数据
版权声明:本文为zhamhzl原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。