pytorch从dataloader中取数据 (python从enumerate或iterator对象中取数据)

报错:

TypeError: 'enumerate' object is not subscriptable
TypeError: 'DataLoader' object is not an iterator

解决代码:

data = torch.utils.data.DataLoader(xxx)
data = iter(data)
(inputs, labels) = next(data)


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部