mmdetection平台下的检测器12次就能训练完了,但是yolov5要训练300次这么多,我的gpu显存24g,yolov5能达到64个batchsize,但是在mmdetection下面16个batchsize就爆了,而且yolov5的每个epoch训练起来都好快,只要几分钟
2条回答 默认 最新
- 爱晚乏客游 2021-04-05 22:49关注
mmdetection没试过,但是yolov5默认300多次是因为人家是在coco数据集上面训练的,而你看yolov5的主页,里面的各个模型的对比可以看出来,基本上都是300epoch的时候达到较好的效果,之后收敛速度变慢,效果不大。如果你训练自己的数据集,你看看loss,P,R和mAP,这些数据达到稳定之后基本上就可以停止了,并不需要300epoch。16个epoch就爆,具体没用过,我猜测估计应该是用空间换时间吧,或者哪里优化没弄好。
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 7无用 3
悬赏问题
- ¥15 metadata提取的PDF元数据,如何转换为一个Excel
- ¥15 关于arduino编程toCharArray()函数的使用
- ¥100 vc++混合CEF采用CLR方式编译报错
- ¥15 coze 的插件输入飞书多维表格 app_token 后一直显示错误,如何解决?
- ¥15 vite+vue3+plyr播放本地public文件夹下视频无法加载
- ¥15 c#逐行读取txt文本,但是每一行里面数据之间空格数量不同
- ¥50 如何openEuler 22.03上安装配置drbd
- ¥20 ING91680C BLE5.3 芯片怎么实现串口收发数据
- ¥15 无线连接树莓派,无法执行update,如何解决?(相关搜索:软件下载)
- ¥15 Windows11, backspace, enter, space键失灵