Update README.md
Browse files
README.md
CHANGED
|
@@ -6,6 +6,20 @@ library_name: timm
|
|
| 6 |
license: mit
|
| 7 |
datasets:
|
| 8 |
- cifar10
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
---
|
| 10 |
# Model card for resnet18_cifar10
|
| 11 |
|
|
@@ -16,6 +30,15 @@ import timm
|
|
| 16 |
model = timm.create_model("hf_hub:SamAdamDay/resnet18_cifar10", pretrained=True)
|
| 17 |
```
|
| 18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
## Model Details
|
| 20 |
- **Dataset:** cifar10
|
| 21 |
- **Number of epochs:** 300
|
|
@@ -24,3 +47,4 @@ model = timm.create_model("hf_hub:SamAdamDay/resnet18_cifar10", pretrained=True)
|
|
| 24 |
- **LR scheduler:** cosine
|
| 25 |
- **PyTorch version:** 2.3.0+cu121
|
| 26 |
- **timm version:** 1.0.7
|
|
|
|
|
|
| 6 |
license: mit
|
| 7 |
datasets:
|
| 8 |
- cifar10
|
| 9 |
+
metrics:
|
| 10 |
+
- accuracy
|
| 11 |
+
model-index:
|
| 12 |
+
- name: resnet18
|
| 13 |
+
results:
|
| 14 |
+
- task:
|
| 15 |
+
type: image-classification
|
| 16 |
+
dataset:
|
| 17 |
+
name: cifar10
|
| 18 |
+
type: cifar10
|
| 19 |
+
metrics:
|
| 20 |
+
- name: accuracy
|
| 21 |
+
type: accuracy
|
| 22 |
+
value: 92.73
|
| 23 |
---
|
| 24 |
# Model card for resnet18_cifar10
|
| 25 |
|
|
|
|
| 30 |
model = timm.create_model("hf_hub:SamAdamDay/resnet18_cifar10", pretrained=True)
|
| 31 |
```
|
| 32 |
|
| 33 |
+
The model was trained using the following command:
|
| 34 |
+
```bash
|
| 35 |
+
./distributed_train.sh --dataset torch/cifar10 --data-dir /root/data --dataset-download --model resnet18 --lr-base 0.3
|
| 36 |
+
```
|
| 37 |
+
|
| 38 |
+
## Metrics
|
| 39 |
+
|
| 40 |
+
The model has a test accuracy of 92.73.
|
| 41 |
+
|
| 42 |
## Model Details
|
| 43 |
- **Dataset:** cifar10
|
| 44 |
- **Number of epochs:** 300
|
|
|
|
| 47 |
- **LR scheduler:** cosine
|
| 48 |
- **PyTorch version:** 2.3.0+cu121
|
| 49 |
- **timm version:** 1.0.7
|
| 50 |
+
|