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