File size: 559 Bytes
29e6b61
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
---
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