xyxing's picture
Create README.md
77f00f8 verified
|
Raw
History Blame Contribute Delete
332 Bytes
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:
phase1 lr = 1e-3, phase2 lr = 4e-5, weight decay = 0.1, batch size = 8, warmup ratio = 0.1
Best validation Macro F1 = 0.8049