【ModelArts】【预置算法】【图像分类-ResNet_v1_50】训练失败
【摘要】 背景：算法来源：图像分类-ResNet_v1_50 (huaweicloud.com)算法名称：图像分类-ResNet_v1_50使用自己的数据去训练，训练失败，界面上提示显存不足可能的原因：1. 显存不足原因：batch_size过大解决办法：逐步减少batch_size2. 图片有问题在日志里面观察到如下错误比如：tensorflow.python.framework.errors_im...
背景：
算法来源：图像分类-ResNet_v1_50 (huaweicloud.com)
算法名称：图像分类-ResNet_v1_50
使用自己的数据去训练，训练失败，界面上提示显存不足
可能的原因：
1. 显存不足
原因：batch_size过大
解决办法：逐步减少batch_size
2. 图片有问题-Invalid JPEG data or crop window
在日志里面观察到如下错误
比如：
tensorflow.python.framework.errors_impl.InvalidArgumentError: Invalid JPEG data or crop window, data size 631
[[{{node map/while/cond/DecodeAndCropJpeg}}]]
[[{{node MultiDeviceIteratorGetNextFromShard}}]]
[[{{node RemoteCall}}]]
[[node IteratorGetNext (defined at /home/work/anaconda/lib/python3.6/site-packages/moxing/tensorflow/executor/learning.py:1322) ]]
[[node global_step/add (defined at /home/work/anaconda/lib/python3.6/site-packages/moxing/tensorflow/executor/learning.py:1174) ]]
原因：部分数据有问题
解决办法：
1. 过滤有问题的数据
2. 使用数据清理
使用数据清洗
【声明】本内容来自华为云开发者社区博主，不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源（华为云社区）、文章链接、文章作者等基本信息，否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
作者其他文章
评论（0）