当前位置:网站首页>【MindSpore】用coco2017训练Model_zoo上的 yolov4,迭代了两千多batch_size之后报错,大佬们帮忙看看。
【MindSpore】用coco2017训练Model_zoo上的 yolov4,迭代了两千多batch_size之后报错,大佬们帮忙看看。
2022-07-30 19:04:00 【小乐快乐】
问题描述:
运行环境:Atlas 800 9000 CANN20.2
batch_size: 4,8,16,32(四种分别尝试过,都在一个epoch训练了80%左右报这个错了)
数据集:coco2017
模型:Model_zoo YOLOv4_Cspdarknet53 (配置都是用的默认的)
解决方案:
从错误截图中初步分析的结论是,数据集可能不完整或是有错误,导致一定step之后再也无法迭代出数据进行训练了。
我们建议按照如下方式排查故障:
1、确保coco2017数据集来源正确,mindspore model_zoo中yolo v4 README提供了coco数据集的官方下载链接:COCO - Common Objects in Context
2、上述错误一般源于src/yolo_dataset.py读取coco数据集时发生了错误,建议采用如下方式进行排查:
在src/yolo_dataset.py的代码底部增加如下调试代码:
from src.config import ConfigYOLOV3DarkNet53
if __name__ == '__main__':
config = ConfigYOLOV3DarkNet53()
config.label_smooth = 0
config.label_smooth_factor = 0.1
image_dir = "/your_path/coco2017/train2017"
anno_path = "/your_path/coco2017/annotations/instances_train2017.json"
yolo_dataset = COCOYoloDataset(image_dir, anno_path)
nums = 0
for data in yolo_dataset:
nums += 1
print("nums is ", nums)
以上代码主要用于调试COCOYoloDataset,正常情况下,yolo_dataset是可以正确完整输出所有数据的,请参照以上调试代码,确保读取coco数据集完整无误。
边栏推荐
- Spark学习:编译Spark项目时遇到的报错
- 浅聊对比学习(Contrastive Learning)第一弹
- VBA批量将Excel数据导入Access数据库
- redis
- Hello, my new name is "Bronze Lock/Tongsuo"
- 设计消息队列存储消息数据的 MySQL 表格
- The large-scale application of artificial intelligence AI products in industrial-grade mature shipping ports of CIMC World Lianda will create a new generation of high-efficiency smart ports and innova
- 生物医学论文有何价值 论文中译英怎样翻译效果好
- VBA 连接Access数据库和Excle
- The problem of writing go run in crontab does not execute
猜你喜欢
OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.解决方法
Golang logging library zerolog use record
开心的聚餐
深入浅出边缘云 | 3. 资源配置
运营 23 年,昔日“国内第一大电商网站”黄了...
Recommendation | People who are kind to you, don't repay them by inviting them to eat
[TypeScript]编译配置
The advanced version of the Niu Ke brushing series (team competition, sorting subsequences, inverting strings, deleting common characters, repairing pastures)
VS Code 连接SQL Server
MySQL data types
随机推荐
设计消息队列存储消息数据的 MySQL 表格
After 23 years of operation, the former "China's largest e-commerce website" has turned yellow...
Immersive experience iFLYTEK 2022 Consumer Expo "Official Designated Product"
Correct pose of Vulkan open feature
Common linked list problems and their Go implementation
Scala学习:类和对象
基于inquirer封装一个控制台文件选择器
Tensorflow2.0 混淆矩阵与打印准确率不符
【PHPWord】PHPOffice 套件之PHPWord快速入门
ctf.show_web5
沉浸式体验科大讯飞2022消博会“官方指定产品”
nlohmann json 使用指南【visual studio 2022】
《痞子衡嵌入式半月刊》 第 59 期
VBA connects Access database and Excel
OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.解决方法
[TypeScript]编译配置
What kind of framework is friendly to developers?
NC | 西湖大学陶亮组-TMPRSS2“助攻”病毒感染并介导索氏梭菌出血毒素的宿主入侵...
谷歌AlphaFold近日宣称预测出地球上几乎所有蛋白质结构
What is the difference between a cloud database and an on-premises database?