【亲测】yoloV8添加Lion优化器

在阅读此博客之前请先熟悉

【亲测】ubuntu20.4利用python脚本训练自己的YOLOV8数据集并推理-CSDN博客

博客里面的步骤意义。

一、在yoloV8环境中安装lion优化器的包

pip install lion-pytorch

二、在yoloV8的trainer.py中导入Lion优化器

from lion_pytorch import Lion

【亲测】yoloV8添加Lion优化器_第1张图片

三、修改trainer.py大约第683行代码添加Lion优化器的判断

elif name == 'Lion':

     optimizer = Lion(g[2], lr=lr, betas=(momentum,0.999), weight_decay=decay)

【亲测】yoloV8添加Lion优化器_第2张图片

四、训练时选择Lion优化器

【亲测】yoloV8添加Lion优化器_第3张图片

你可能感兴趣的:(1024程序员节)