YOLOV5模型训练

之前在博文中讲到了YOLOV5的运行,以及转tensorrt.

但是, 一个模型通常需要结合数据训练,才能得到更好的结果. 因此,我们有必要熟悉yolov5的训练过程. 

执行训练的过程

Yolov5的github提供了官方的训练脚本. 

 第一次运行,会自动下载数据集,然后会检测到你的gpu配置,如果不对,那么它会自动退出. 因此, 请确保你的电脑有GPU, 并且, 适配了cuda和pytorch. 

相关的检验可以参考这篇博文

AssertionError: Torch not compiled with CUDA enabled解決方案_海滩油炸的博客-CSDN博客

开始进行训练后, 请注意设定batch-size, 第一次使用2080Ti训练的时候, batch-size设定为128, 内存直接爆炸.

然后重启电脑后, 根据官方建议,batch_size设定为 --batch-size -1 ,训练总算开始了.

 如上图截图可以看到

* 网络的参数总结(layers, paramters, gradients, GFLOP

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值