YOLOv6_pro icon indicating copy to clipboard operation
YOLOv6_pro copied to clipboard

Published 20 hours ago • verified publisher icon yang-0201

如何使用cache将训练图片放入内存？我将datasets所有cache=True似乎没有什么变化

Open liiicon opened this issue 1 year ago • 3 comments

liiicon avatar Oct 16 '23 12:10 liiicon

在这个版本里没有支持用cache载入训练图片，可以尝试自己添加，在官方版本里已经支持了，改动代码如下：https://github.com/meituan/YOLOv6/pull/924/files

yang-0201 avatar Oct 16 '23 12:10 yang-0201

感谢，每轮 的训练时间有减少1min左右，但是每轮评估map时还是很慢。这是使用cache之前 980c8cad5cecc01bf3205605c01c8bf这是使用cache之后 87a1e58c4beb3cdefa527cc9be1a039 我用v7训练时，每轮训练后评估map推理时间都在1min之下。

liiicon avatar Oct 17 '23 14:10 liiicon

训练会加快就好！验证的话，亲测这个版本的v6是存在前面几轮验证很久的情况，后面的轮数会快很多～

yang-0201 avatar Oct 18 '23 07:10 yang-0201