So i want to take a yolov8 classification model, freeze the layers and train it for a multilabel classification task, i changed the last layer and added a sigmoid activation function. but when i try to fit the model on my dataset it trains on imagenet instead.
is there a specific parameter that turns off this behaviour?
here is a part of my code.
”’
from ultralytics import YOLO
import torch.nn as nn