在阅读此博客之前请先熟悉
【亲测】ubuntu20.4利用python脚本训练自己的YOLOV8数据集并推理-CSDN博客
博客里面的步骤意义。
一、在yoloV8环境中安装lion优化器的包
pip install lion-pytorch
二、在yoloV8的trainer.py中导入Lion优化器
from lion_pytorch import Lion
三、修改trainer.py大约第683行代码添加Lion优化器的判断
elif name == 'Lion':
optimizer = Lion(g[2], lr=lr, betas=(momentum,0.999), weight_decay=decay)
四、训练时选择Lion优化器