当前位置:网站首页>【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数据集完整无误。
边栏推荐
- 【剑指 Offe】剑指 Offer 17. 打印从1到最大的n位数
- [Summary] 1396- 60+ VSCode plugins to create a useful editor
- Codeblocks + Widgets create window code analysis
- 自己需要努力
- 浅聊对比学习(Contrastive Learning)第一弹
- 生物医学论文有何价值 论文中译英怎样翻译效果好
- Pytorch foundation -- tensorboard use (1)
- Critical Reviews | 南农邹建文组综述全球农田土壤抗生素与耐药基因分布
- LeetCode Exercise - Two Questions About Finding Sum of Array Elements
- golang日志库zerolog使用记录
猜你喜欢

golang日志库zerolog使用记录

VBA connects Access database and Excel

沉浸式体验科大讯飞2022消博会“官方指定产品”

nlohmann json 使用指南【visual studio 2022】

Delay queue optimization (2)

Mysql execution principle analysis

What is the value of biomedical papers? How to translate the papers into Chinese and English?

Deepen school-enterprise cooperation and build an "overpass" for the growth of technical and skilled talents
![【Prometheus】Prometheus联邦的一次优化记录[续]](/img/5d/56e171b7a02584337a0cfe5c731fb2.png)
【Prometheus】Prometheus联邦的一次优化记录[续]

The advanced version of the Niu Ke brushing series (team competition, sorting subsequences, inverting strings, deleting common characters, repairing pastures)
随机推荐
crontab中写go run不执行的问题
MySQL data types
AI基础:图解Transformer
node封装一个控制台进度条插件
nlohmann json 使用指南【visual studio 2022】
Mysql执行原理剖析
【每日一道LeetCode】——191. 位1的个数
The use of terminal split screen tool Terminalx
Golang logging library zerolog use record
中集世联达飞瞳全球工业人工智能AI领军者,全球顶尖AI核心技术高泛化性高鲁棒性稀疏样本持续学习,工业级高性能成熟AI产品规模应用
监听开机广播
7.29模拟赛总结
Chapter 14 Type Information
Scrapy framework is introduced
ctf.show_web5
谷歌AlphaFold近日宣称预测出地球上几乎所有蛋白质结构
LocalDate时间生成
golang日志库zerolog使用记录
Anaconda Navigator stuck on loading applications
自然语言处理nltk