当前位置:网站首页>【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数据集完整无误。
边栏推荐
- VBA 连接Access数据库和Excle
- 二分答案裸题(加一点鸽巢原理)
- The use of terminal split screen tool Terminalx
- 2种手绘风格效果比较,你更喜欢哪一种呢?
- WEBSOCKETPP使用简介+demo
- Immersive experience iFLYTEK 2022 Consumer Expo "Official Designated Product"
- 中集世联达飞瞳全球工业人工智能AI领军者,全球顶尖AI核心技术高泛化性高鲁棒性稀疏样本持续学习,工业级高性能成熟AI产品规模应用
- 【Pointing to Offer】Pointing to Offer 22. The kth node from the bottom in the linked list
- 阿里云武林头条活动分享
- VBA batch import Excel data into Access database
猜你喜欢
运营 23 年,昔日“国内第一大电商网站”黄了...
中集世联达飞瞳全球工业人工智能AI领军者,全球顶尖AI核心技术高泛化性高鲁棒性稀疏样本持续学习,工业级高性能成熟AI产品规模应用
Does the satellite phone communicate directly with the satellite or through a ground station?
MySql中@符号的使用
Scrapy framework is introduced
Pytorch foundation -- tensorboard use (1)
Deepen school-enterprise cooperation and build an "overpass" for the growth of technical and skilled talents
Node encapsulates a console progress bar plugin
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
谷歌AlphaFold近日宣称预测出地球上几乎所有蛋白质结构
随机推荐
几个GTest、GMock的例子
VS Code 连接SQL Server
Correct pose of Vulkan open feature
Recommendation | People who are kind to you, don't repay them by inviting them to eat
【总结】1396- 60+个 VSCode 插件,打造好用的编辑器
《痞子衡嵌入式半月刊》 第 59 期
MongoDB打破了原则引入SQL?
Scrapy framework is introduced
crontab中写go run不执行的问题
LocalDate时间生成
Witness the magical awakening of the mini world in HUAWEI CLOUD
实体中增加操作方法
6块钱1斤,日本公司为何来中国收烟头?
SwiftUI iOS 精品开源项目之 完整烘焙食品菜谱App基于SQLite(教程含源码)
requet.getHeader("token") is null
Critical Reviews | 南农邹建文组综述全球农田土壤抗生素与耐药基因分布
VBA connects Access database and Excel
高精度加法
AWS console
Go 系统收集