許多時候我們會有一些對模型結(jié)構(gòu)進(jìn)行修改的需求，這個時候如何進(jìn)行模型改進(jìn)就成了我們的問題，接下來我的流程中將會介紹如何替換YOLOv8的優(yōu)化器。
1. 首先準(zhǔn)備代碼
比如我要將YOLOv8中的優(yōu)化器替換為Lion，則需要準(zhǔn)備Lion的pytorch代碼，具體的地址為https://github.com/google/automl/tree/master/lion，其中有一個lion_pytorch.py，即我們所需要的代碼文件。
2. 導(dǎo)入優(yōu)化器
將lion_pytorch.py文件放置在yolov8項目中，比如我用conda安裝的yolov8，則將其放置在miniconda3/envs/yolo8/lib/python3.8/site-packages/ultralytics/engine文件夾中，同時，在訓(xùn)練器的文件trainer.py中進(jìn)行導(dǎo)入，首先，添加圖1中的代碼，代碼前面的是文件中的行數(shù)。
圖1
再在代碼中添加圖2中的代碼
圖2
至此，即完成了優(yōu)化器的導(dǎo)入。
3. 替換優(yōu)化器
有兩種方法，分別是在miniconda3/envs/yolo8/lib/python3.8/site-packages/ultralytics/cfg/default.yaml中替換和在訓(xùn)練時直接指定optimizer為Lion，如果要在default.yaml中進(jìn)行替換，則是在如圖3所示的地方進(jìn)行替換，將auto替換為Lion
圖3
另一種方法則是在train.py指定優(yōu)化器，如圖4所示。
圖4
如此，便實現(xiàn)了優(yōu)化器的替換。
