baseline.py

from transformers import ConvNextV2ForImageClassification

def convnext_v2_base(num_classes):
    model = ConvNextV2ForImageClassification.from_pretrained(
        "facebook/convnextv2-base-22k-224",
        num_labels=num_classes,
        ignore_mismatched_sizes=True 
    )
    return model