fullstuck commited on
Commit
eb42d92
1 Parent(s): b8fac34

End of training

Browse files
Files changed (3) hide show
  1. README.md +8 -75
  2. pytorch_model.bin +2 -2
  3. training_args.bin +2 -2
README.md CHANGED
@@ -5,24 +5,9 @@ tags:
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
8
- metrics:
9
- - accuracy
10
  model-index:
11
  - name: image_classification
12
- results:
13
- - task:
14
- name: Image Classification
15
- type: image-classification
16
- dataset:
17
- name: imagefolder
18
- type: imagefolder
19
- config: en-US
20
- split: train
21
- args: en-US
22
- metrics:
23
- - name: Accuracy
24
- type: accuracy
25
- value: 0.525
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +17,12 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 1.9905
36
- - Accuracy: 0.525
 
 
 
 
37
 
38
  ## Model description
39
 
@@ -52,7 +41,7 @@ More information needed
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
- - learning_rate: 0.0001
56
  - train_batch_size: 16
57
  - eval_batch_size: 16
58
  - seed: 42
@@ -60,62 +49,6 @@ The following hyperparameters were used during training:
60
  - lr_scheduler_type: linear
61
  - num_epochs: 50
62
 
63
- ### Training results
64
-
65
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
- | No log | 1.0 | 40 | 1.7409 | 0.2875 |
68
- | No log | 2.0 | 80 | 1.5124 | 0.4375 |
69
- | No log | 3.0 | 120 | 1.4255 | 0.4437 |
70
- | No log | 4.0 | 160 | 1.4154 | 0.425 |
71
- | No log | 5.0 | 200 | 1.2886 | 0.4625 |
72
- | No log | 6.0 | 240 | 1.2963 | 0.5125 |
73
- | No log | 7.0 | 280 | 1.3139 | 0.55 |
74
- | No log | 8.0 | 320 | 1.2976 | 0.5312 |
75
- | No log | 9.0 | 360 | 1.4368 | 0.5062 |
76
- | No log | 10.0 | 400 | 1.4022 | 0.5062 |
77
- | No log | 11.0 | 440 | 1.2853 | 0.55 |
78
- | No log | 12.0 | 480 | 1.3265 | 0.5563 |
79
- | 0.8831 | 13.0 | 520 | 1.3894 | 0.55 |
80
- | 0.8831 | 14.0 | 560 | 1.4465 | 0.5312 |
81
- | 0.8831 | 15.0 | 600 | 1.7185 | 0.475 |
82
- | 0.8831 | 16.0 | 640 | 1.7408 | 0.4875 |
83
- | 0.8831 | 17.0 | 680 | 1.5199 | 0.5437 |
84
- | 0.8831 | 18.0 | 720 | 1.7238 | 0.525 |
85
- | 0.8831 | 19.0 | 760 | 1.8348 | 0.4875 |
86
- | 0.8831 | 20.0 | 800 | 1.6278 | 0.5125 |
87
- | 0.8831 | 21.0 | 840 | 1.7539 | 0.5 |
88
- | 0.8831 | 22.0 | 880 | 1.9007 | 0.4938 |
89
- | 0.8831 | 23.0 | 920 | 1.6903 | 0.5375 |
90
- | 0.8831 | 24.0 | 960 | 1.7954 | 0.5062 |
91
- | 0.2214 | 25.0 | 1000 | 1.7070 | 0.575 |
92
- | 0.2214 | 26.0 | 1040 | 1.6764 | 0.5625 |
93
- | 0.2214 | 27.0 | 1080 | 1.8590 | 0.5188 |
94
- | 0.2214 | 28.0 | 1120 | 1.7531 | 0.5188 |
95
- | 0.2214 | 29.0 | 1160 | 1.5238 | 0.5875 |
96
- | 0.2214 | 30.0 | 1200 | 1.6463 | 0.6 |
97
- | 0.2214 | 31.0 | 1240 | 1.7955 | 0.5563 |
98
- | 0.2214 | 32.0 | 1280 | 1.9920 | 0.5 |
99
- | 0.2214 | 33.0 | 1320 | 1.8826 | 0.55 |
100
- | 0.2214 | 34.0 | 1360 | 2.0573 | 0.5 |
101
- | 0.2214 | 35.0 | 1400 | 1.8438 | 0.5312 |
102
- | 0.2214 | 36.0 | 1440 | 1.9004 | 0.5312 |
103
- | 0.2214 | 37.0 | 1480 | 1.8215 | 0.5437 |
104
- | 0.1479 | 38.0 | 1520 | 2.0467 | 0.5437 |
105
- | 0.1479 | 39.0 | 1560 | 1.8564 | 0.5687 |
106
- | 0.1479 | 40.0 | 1600 | 1.8381 | 0.5687 |
107
- | 0.1479 | 41.0 | 1640 | 1.8110 | 0.5687 |
108
- | 0.1479 | 42.0 | 1680 | 2.0587 | 0.5375 |
109
- | 0.1479 | 43.0 | 1720 | 1.9597 | 0.5687 |
110
- | 0.1479 | 44.0 | 1760 | 1.9199 | 0.5687 |
111
- | 0.1479 | 45.0 | 1800 | 1.8714 | 0.5312 |
112
- | 0.1479 | 46.0 | 1840 | 1.9463 | 0.575 |
113
- | 0.1479 | 47.0 | 1880 | 2.0449 | 0.5312 |
114
- | 0.1479 | 48.0 | 1920 | 1.9172 | 0.525 |
115
- | 0.1479 | 49.0 | 1960 | 1.9153 | 0.55 |
116
- | 0.0946 | 50.0 | 2000 | 1.9905 | 0.525 |
117
-
118
-
119
  ### Framework versions
120
 
121
  - Transformers 4.33.2
 
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
 
 
8
  model-index:
9
  - name: image_classification
10
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ---
12
 
13
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
19
  It achieves the following results on the evaluation set:
20
+ - eval_loss: 2.0725
21
+ - eval_accuracy: 0.1437
22
+ - eval_runtime: 142.7817
23
+ - eval_samples_per_second: 1.121
24
+ - eval_steps_per_second: 0.07
25
+ - step: 0
26
 
27
  ## Model description
28
 
 
41
  ### Training hyperparameters
42
 
43
  The following hyperparameters were used during training:
44
+ - learning_rate: 5e-06
45
  - train_batch_size: 16
46
  - eval_batch_size: 16
47
  - seed: 42
 
49
  - lr_scheduler_type: linear
50
  - num_epochs: 50
51
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
52
  ### Framework versions
53
 
54
  - Transformers 4.33.2
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed5779361f0de45f8075c1798a690c61197e57b31cc0af2346832814c9d5d670
3
- size 343287149
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da85c61d6a8b1d6bddacd30e6747c77b3786a8d9dbffaafa4a3d69ac9a52b134
3
+ size 343284397
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5ccddbfbc0d2415e1b7ea870fa5bdca1b71c03f582b362459b39b95d72957a3a
3
- size 4091
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5d79b1ddb120af0a79f19df3a9bf9fac441952fc4da887b8c89ed2536f50e97
3
+ size 4027