| | --- |
| | 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 |
