linkanjarad commited on
Commit
e1a3fb7
1 Parent(s): 376ca92

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -11
README.md CHANGED
@@ -21,7 +21,7 @@ model-index:
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
- value: 0.7145092460881934
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,8 +31,8 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  This model is a fine-tuned version of [google/mobilenet_v2_1.0_224](https://huggingface.co/google/mobilenet_v2_1.0_224) on the image_folder dataset.
33
  It achieves the following results on the evaluation set:
34
- - Loss: 1.3799
35
- - Accuracy: 0.7145
36
 
37
  ## Model description
38
 
@@ -57,23 +57,26 @@ The following hyperparameters were used during training:
57
  - seed: 42
58
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
59
  - lr_scheduler_type: linear
60
- - lr_scheduler_warmup_ratio: 0.1
61
- - num_epochs: 5
62
 
63
  ### Training results
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
- | 2.3889 | 1.0 | 248 | 2.1952 | 0.6310 |
68
- | 1.8202 | 2.0 | 496 | 1.6363 | 0.7013 |
69
- | 1.6266 | 3.0 | 744 | 1.6291 | 0.6343 |
70
- | 1.5566 | 4.0 | 992 | 1.3514 | 0.7129 |
71
- | 1.5507 | 5.0 | 1240 | 1.3799 | 0.7145 |
 
 
 
72
 
73
 
74
  ### Framework versions
75
 
76
- - Transformers 4.26.1
77
  - Pytorch 1.13.0
78
  - Datasets 2.1.0
79
  - Tokenizers 0.13.2
 
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
+ value: 0.7857752489331437
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 [google/mobilenet_v2_1.0_224](https://huggingface.co/google/mobilenet_v2_1.0_224) on the image_folder dataset.
33
  It achieves the following results on the evaluation set:
34
+ - Loss: 1.0461
35
+ - Accuracy: 0.7858
36
 
37
  ## Model description
38
 
 
57
  - seed: 42
58
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
59
  - lr_scheduler_type: linear
60
+ - lr_scheduler_warmup_ratio: 0.15
61
+ - num_epochs: 8
62
 
63
  ### Training results
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
+ | 2.9265 | 1.0 | 248 | 2.7159 | 0.4703 |
68
+ | 1.9734 | 2.0 | 496 | 1.7668 | 0.6649 |
69
+ | 1.7206 | 3.0 | 744 | 1.4012 | 0.7206 |
70
+ | 1.6406 | 4.0 | 992 | 1.2514 | 0.7644 |
71
+ | 1.6075 | 5.0 | 1240 | 1.2934 | 0.7094 |
72
+ | 1.5932 | 6.0 | 1488 | 1.2093 | 0.7257 |
73
+ | 1.5203 | 7.0 | 1736 | 1.0461 | 0.7858 |
74
+ | 1.5076 | 8.0 | 1984 | 1.0580 | 0.7848 |
75
 
76
 
77
  ### Framework versions
78
 
79
+ - Transformers 4.27.3
80
  - Pytorch 1.13.0
81
  - Datasets 2.1.0
82
  - Tokenizers 0.13.2