How to use Ranger optimizer with YOLOV8?

[triciardvn]

This is the error for my code since I want to try the "Ranger" optimizer instead of the existing available ones using YOLOV8 (in google colab).

NotImplementedError: Optimizer 'Ranger' not found in list of available optimizers [Adam, AdamW, NAdam, RAdam, RMSProp, SGD, auto].To request support for addition optimizers please visit https://github.com/ultralytics/ultralytics.

How can I integrate this new optimizer as I want to do an object detection for etrikes and ebikes (I want to start training it). I already tried this code:

!pip install git+https://github.com/lessw2020/Ranger21.git

from ultralytics import YOLO
import os

# Load your model
model = YOLO('yolov8n.pt')

# Define hyperparameters
hyperparameters = {
    'lr': 0.001,
    'weight_decay': 0.0005,
    'momentum': 0.937
}

# Get the number of training images
num_train_images = len(os.listdir('/content/drive/MyDrive/e/train/images'))
batch_size = 16
num_batches_per_epoch = num_train_images // batch_size
num_epochs = 10

# Setup training
model.train(data='/content/drive/MyDrive/e/data.yaml', epochs=num_epochs, imgsz=640, batch=batch_size, optimizer='Ranger')

Please let me know what to do ASAP. Thank you.

Note: I need to use the ranger21 optimizer as I need it for my thesis.