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

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