YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

Fine-tuned google/vit-base-patch16-224 for 7-class skin lesion classification on HAM10000. Part of the DermAI explainability project comparing CNN and Vision Transformer explanations.

Hyperparameter tuned: dropout=0.2, phase1 lr = 5e-2, phase2 lr = 4e-5, weight decay = 0.1, batch size = 8, warmup ratio = 0.1, no augmentation

Best validation Macro F1 = 0.8074

Downloads last month: 15

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support