metadata
license: apache-2.0
base_model:
- microsoft/resnet-18
ResNet model card (MedMNIST – OrganAMNIST)
Dataset
- Collection: MedMNIST , task: OrganAMNIST (multi-class, 11 organs).
- Image format: grayscale, 28×28 resized to 224×224 via
MedMNIST_dataset. - Available samples: train 34,561; val 6,491; test 17,778.
- Used for this run: random subset of 10,000 train images; full val set.
Training setup
- Architecture: ResNet18
- Epochs: 10
- Batch size: 64
- Optimizer: Adam (lr 0.001, weight_decay 0.0001)
- Num workers: 4
- Device: CUDA