训练Pytorch深度学习模型出现StopIteration
训练Pytorch深度学习模型出现StopIteration
·
训练一个深度学习检测模型,突然出现:
是因为next(batch_iterator),可能迭代器读出来的数据为空。
# load train data
# 原先代码
images, targets = next(batch_iterator)
# 更改为:
try:
images, targets = next(batch_iterator)
except:
batch_iterator = iter(data_loader)
images, targets = next(batch_iterator)
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐

所有评论(0)