qyu21490 2021-04-05 20:56 采纳率: 50%
浏览 8906
已结题

yolov5为什么默认要训练300次这么多

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就爆,具体没用过,我猜测估计应该是用空间换时间吧,或者哪里优化没弄好。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 11月23日
  • 已采纳回答 11月15日

悬赏问题

  • ¥500 高有偿提问!求优化设计微信小程序
  • ¥15 matlab在安装时报错 无法找到入口 无法定位程序输入点
  • ¥15 收益高的广告联盟有哪些
  • ¥15 Android Studio webview 的使用问题, 播放器横屏全屏
  • ¥15 删掉jdk后重新下载,Java web所需要的eclipse无法使用
  • ¥15 uniapp正式环境中通过webapi将本地数据推送到设备出现的跨域问题
  • ¥15 xui建立节点,显示错误
  • ¥15 关于#单片机#的问题:开始、复位、十进制的功能可以实现,但是切换八进制的功能无法实现(按下按键也没有效果),把初始状态调成八进制,也是八进制可以实现但是切换到十进制不行(相关搜索:汇编语言|计数器)
  • ¥15 VINS-Mono或Fusion中feature_manager中estimated_depth是特征的深度还是逆深度?
  • ¥15 谷歌浏览器如何备份抖音网页数据