Andru commited on
Commit
a2f0d16
1 Parent(s): e35babd

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -10
README.md CHANGED
@@ -1,6 +1,5 @@
1
  ---
2
  license: apache-2.0
3
- base_model: microsoft/swin-tiny-patch4-window7-224
4
  tags:
5
  - generated_from_trainer
6
  datasets:
@@ -22,7 +21,7 @@ model-index:
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
- value: 0.9840740740740741
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +31,8 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.0557
36
- - Accuracy: 0.9841
37
 
38
  ## Model description
39
 
@@ -67,14 +66,14 @@ The following hyperparameters were used during training:
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
- | 0.2386 | 1.0 | 190 | 0.0965 | 0.9693 |
71
- | 0.1616 | 2.0 | 380 | 0.0751 | 0.9741 |
72
- | 0.1469 | 3.0 | 570 | 0.0557 | 0.9841 |
73
 
74
 
75
  ### Framework versions
76
 
77
- - Transformers 4.33.1
78
- - Pytorch 2.0.1+cu118
79
- - Datasets 2.14.5
80
  - Tokenizers 0.13.3
 
1
  ---
2
  license: apache-2.0
 
3
  tags:
4
  - generated_from_trainer
5
  datasets:
 
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
+ value: 0.9358974358974359
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
 
32
  This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on the imagefolder dataset.
33
  It achieves the following results on the evaluation set:
34
+ - Loss: 0.1772
35
+ - Accuracy: 0.9359
36
 
37
  ## Model description
38
 
 
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
69
+ | 0.2447 | 0.99 | 98 | 0.1806 | 0.9323 |
70
+ | 0.1986 | 1.99 | 197 | 0.1772 | 0.9359 |
71
+ | 0.1933 | 2.98 | 294 | 0.1772 | 0.9359 |
72
 
73
 
74
  ### Framework versions
75
 
76
+ - Transformers 4.30.0
77
+ - Pytorch 2.2.1+cu121
78
+ - Datasets 2.18.0
79
  - Tokenizers 0.13.3