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

Check out the documentation for more information.

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

Hyperparameter tuned: phase1 lr = 4e-4, phase2 lr = 1e-4, weight decay = 0.01, batch size = 32, warmup ratio = 0.1

Best validation Macro F1 = 0.7073

Downloads last month
51
Safetensors
Model size
4.06M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support