enumerate和iter的使用

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)函数获取可迭代对象的数据

你可能感兴趣的:(计算机视觉,神经网络,深度学习,python,pytorch)