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