Hugging Face's logo

tiennguyenbnbk
/
teacher-status-van-tiny-256

Image Classification
Transformers
TensorBoard
Safetensors
van
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Metrics Community
Nguyen Tien commited on
Commit
13e9bbb
•
1 Parent(s): 6462e67

Model save

Browse files
Files changed (2) hide show
  1. README.md +65 -12
  2. model.safetensors +1 -1
README.md CHANGED
@@ -5,9 +5,32 @@ tags:
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
 
 
 
 
8
  model-index:
9
  - name: teacher-status-van-tiny-256
10
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ---
12
 
13
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,16 +40,11 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [Visual-Attention-Network/van-tiny](https://huggingface.co/Visual-Attention-Network/van-tiny) on the imagefolder dataset.
19
  It achieves the following results on the evaluation set:
20
- - eval_loss: 0.2176
21
- - eval_accuracy: 0.9213
22
- - eval_f1_score: 0.9307
23
- - eval_recall: 0.9307
24
- - eval_precision: 0.9307
25
- - eval_runtime: 1.2169
26
- - eval_samples_per_second: 146.275
27
- - eval_steps_per_second: 4.931
28
- - epoch: 11.28
29
- - step: 141
30
 
31
  ## Model description
32
 
@@ -45,7 +63,7 @@ More information needed
45
  ### Training hyperparameters
46
 
47
  The following hyperparameters were used during training:
48
- - learning_rate: 0.0001
49
  - train_batch_size: 32
50
  - eval_batch_size: 32
51
  - seed: 42
@@ -56,6 +74,41 @@ The following hyperparameters were used during training:
56
  - lr_scheduler_warmup_ratio: 0.1
57
  - num_epochs: 30
58
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
59
  ### Framework versions
60
 
61
  - Transformers 4.36.2
 
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
8
+ metrics:
9
+ - accuracy
10
+ - recall
11
+ - precision
12
  model-index:
13
  - name: teacher-status-van-tiny-256
14
+ results:
15
+ - task:
16
+ name: Image Classification
17
+ type: image-classification
18
+ dataset:
19
+ name: imagefolder
20
+ type: imagefolder
21
+ config: default
22
+ split: train
23
+ args: default
24
+ metrics:
25
+ - name: Accuracy
26
+ type: accuracy
27
+ value: 0.949438202247191
28
+ - name: Recall
29
+ type: recall
30
+ value: 0.9473684210526315
31
+ - name: Precision
32
+ type: precision
33
+ value: 0.9574468085106383
34
  ---
35
 
36
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
40
 
41
  This model is a fine-tuned version of [Visual-Attention-Network/van-tiny](https://huggingface.co/Visual-Attention-Network/van-tiny) on the imagefolder dataset.
42
  It achieves the following results on the evaluation set:
43
+ - Loss: 0.1130
44
+ - Accuracy: 0.9494
45
+ - F1 Score: 0.9524
46
+ - Recall: 0.9474
47
+ - Precision: 0.9574
 
 
 
 
 
48
 
49
  ## Model description
50
 
 
63
  ### Training hyperparameters
64
 
65
  The following hyperparameters were used during training:
66
+ - learning_rate: 5e-05
67
  - train_batch_size: 32
68
  - eval_batch_size: 32
69
  - seed: 42
 
74
  - lr_scheduler_warmup_ratio: 0.1
75
  - num_epochs: 30
76
 
77
+ ### Training results
78
+
79
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Score | Recall | Precision |
80
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|:------:|:---------:|
81
+ | 0.6928 | 0.96 | 12 | 0.6904 | 0.6685 | 0.7631 | 1.0 | 0.6169 |
82
+ | 0.6893 | 2.0 | 25 | 0.6683 | 0.5393 | 0.6985 | 1.0 | 0.5367 |
83
+ | 0.6726 | 2.96 | 37 | 0.5704 | 0.5843 | 0.7197 | 1.0 | 0.5621 |
84
+ | 0.5295 | 4.0 | 50 | 0.4148 | 0.9213 | 0.9263 | 0.9263 | 0.9263 |
85
+ | 0.4745 | 4.96 | 62 | 0.3108 | 0.9382 | 0.9430 | 0.9579 | 0.9286 |
86
+ | 0.4206 | 6.0 | 75 | 0.2301 | 0.9438 | 0.9474 | 0.9474 | 0.9474 |
87
+ | 0.3898 | 6.96 | 87 | 0.1820 | 0.9494 | 0.9519 | 0.9368 | 0.9674 |
88
+ | 0.3153 | 8.0 | 100 | 0.1545 | 0.9494 | 0.9538 | 0.9789 | 0.93 |
89
+ | 0.3077 | 8.96 | 112 | 0.1521 | 0.9607 | 0.9622 | 0.9368 | 0.9889 |
90
+ | 0.3048 | 10.0 | 125 | 0.1331 | 0.9607 | 0.9626 | 0.9474 | 0.9783 |
91
+ | 0.3004 | 10.96 | 137 | 0.1314 | 0.9607 | 0.9634 | 0.9684 | 0.9583 |
92
+ | 0.2839 | 12.0 | 150 | 0.1272 | 0.9607 | 0.9622 | 0.9368 | 0.9889 |
93
+ | 0.286 | 12.96 | 162 | 0.1189 | 0.9607 | 0.9622 | 0.9368 | 0.9889 |
94
+ | 0.2473 | 14.0 | 175 | 0.0977 | 0.9719 | 0.9733 | 0.9579 | 0.9891 |
95
+ | 0.2774 | 14.96 | 187 | 0.0988 | 0.9831 | 0.9841 | 0.9789 | 0.9894 |
96
+ | 0.2541 | 16.0 | 200 | 0.0969 | 0.9719 | 0.9733 | 0.9579 | 0.9891 |
97
+ | 0.2383 | 16.96 | 212 | 0.1042 | 0.9719 | 0.9733 | 0.9579 | 0.9891 |
98
+ | 0.2552 | 18.0 | 225 | 0.1081 | 0.9719 | 0.9733 | 0.9579 | 0.9891 |
99
+ | 0.2223 | 18.96 | 237 | 0.1150 | 0.9663 | 0.9681 | 0.9579 | 0.9785 |
100
+ | 0.2561 | 20.0 | 250 | 0.1234 | 0.9551 | 0.9574 | 0.9474 | 0.9677 |
101
+ | 0.2462 | 20.96 | 262 | 0.1178 | 0.9607 | 0.9630 | 0.9579 | 0.9681 |
102
+ | 0.2294 | 22.0 | 275 | 0.1262 | 0.9382 | 0.9430 | 0.9579 | 0.9286 |
103
+ | 0.2296 | 22.96 | 287 | 0.1290 | 0.9438 | 0.9479 | 0.9579 | 0.9381 |
104
+ | 0.2224 | 24.0 | 300 | 0.1153 | 0.9494 | 0.9529 | 0.9579 | 0.9479 |
105
+ | 0.2205 | 24.96 | 312 | 0.1150 | 0.9494 | 0.9529 | 0.9579 | 0.9479 |
106
+ | 0.2169 | 26.0 | 325 | 0.1121 | 0.9551 | 0.9574 | 0.9474 | 0.9677 |
107
+ | 0.2212 | 26.96 | 337 | 0.1145 | 0.9494 | 0.9529 | 0.9579 | 0.9479 |
108
+ | 0.2188 | 28.0 | 350 | 0.1131 | 0.9494 | 0.9524 | 0.9474 | 0.9574 |
109
+ | 0.2015 | 28.8 | 360 | 0.1130 | 0.9494 | 0.9524 | 0.9474 | 0.9574 |
110
+
111
+
112
  ### Framework versions
113
 
114
  - Transformers 4.36.2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4120006479559a69e64377e213d1a39dd0045a9fa7c2698ca889142fc875dec6
3
  size 15480696
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68955bac45fc41041d0a6b2343cca4287964da18e94b1f66737f609350876d10
3
  size 15480696