Yolov8+Lion优化器

本文介绍了如何在Yolov8中整合Lion优化器,包括下载配置、导入LionPyTorch库、在trainer源码中应用以及使用Lion训练模型。作者分享了在实践中遇到的问题和提示,如可能的小批量训练效果不佳。
摘要由CSDN通过智能技术生成


1.下载并配置yolov8

这里对于yolov8的detect和train的基本流程不做描述。
参考gitube地址:yolov8源码地址

2.导入Lion优化器

Lion
2023年2月,Google 和 UCLA 提出 Lion (EvoLved Sign Momentum) 优化算法,以优越的性能和良好的效果超越了经典的优化算法。Lion 通过程序搜索来发现优化算法,并且用其改进深度神经网络训练。

导入
在python终端中输入以下内容,导入Lion优化器

$ pip install lion-pytorch

参考gitube地址:lion-pytorch地址


3.在trainer源码中引入Lion优化器

项目目录中找到trainer.py
在这里插入图片描述
导入相应的包

from lion_pytorch import Lion

在这里插入图片描述
找到图中相应位置,添加相应代码。

elif name == 'Lion':
    optimizer = Lion(params=g[2], lr=lr, betas=(momentum, 0.999), weight_decay=decay)

在这里插入图片描述
此处的导入方式同样参考的是
gitube地址:lion-pytorch地址


4.使用Lion优化器训练模型

找到yolo目录下的default.yaml文件在这里插入图片描述
改变optimizer为Lion,然后开始训练模型即可。
在这里插入图片描述


Tips

  1. 本人使用coco数据集时,并没有得到比SGD等传统优化器好的结果,可能的原因是,虽然 Lion 有更强的鲁棒性和泛化能力,但它在小的 batch_size 时,表现不好,或者是本人导包导错了。
  2. 本人也处于学习阶段,不确定导入方式是否正确,欢迎大家一起交流。
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值