训练一个深度学习检测模型,突然出现:
在这里插入图片描述
是因为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)
Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐