SamAdamDay commited on
Commit
13f3a26
·
verified ·
1 Parent(s): 2596b33

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
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
+