MattyB95 commited on
Commit
11c0b65
1 Parent(s): 98b9645

End of training

Browse files
README.md CHANGED
@@ -25,16 +25,16 @@ model-index:
25
  metrics:
26
  - name: Accuracy
27
  type: accuracy
28
- value: 0.9996377394944453
29
  - name: F1
30
  type: f1
31
- value: 0.9997981836528759
32
  - name: Precision
33
  type: precision
34
- value: 0.9997309296381004
35
  - name: Recall
36
  type: recall
37
- value: 0.9998654467168999
38
  ---
39
 
40
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -44,11 +44,11 @@ should probably proofread and complete it, then remove this comment. -->
44
 
45
  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.
46
  It achieves the following results on the evaluation set:
47
- - Loss: 0.0025
48
- - Accuracy: 0.9996
49
- - F1: 0.9998
50
- - Precision: 0.9997
51
- - Recall: 0.9999
52
 
53
  ## Model description
54
 
 
25
  metrics:
26
  - name: Accuracy
27
  type: accuracy
28
+ value: 0.7166781307466625
29
  - name: F1
30
  type: f1
31
+ value: 0.8124204206436981
32
  - name: Precision
33
  type: precision
34
+ value: 0.9998169964543063
35
  - name: Recall
36
  type: recall
37
+ value: 0.6841833380294918
38
  ---
39
 
40
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
44
 
45
  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.
46
  It achieves the following results on the evaluation set:
47
+ - Loss: 2.0649
48
+ - Accuracy: 0.7167
49
+ - F1: 0.8124
50
+ - Precision: 0.9998
51
+ - Recall: 0.6842
52
 
53
  ## Model description
54
 
all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.7166781307466625,
4
+ "eval_f1": 0.8124204206436981,
5
+ "eval_loss": 2.0649054050445557,
6
+ "eval_precision": 0.9998169964543063,
7
+ "eval_recall": 0.6841833380294918,
8
+ "eval_runtime": 1515.2798,
9
+ "eval_samples_per_second": 47.012,
10
+ "eval_steps_per_second": 5.877,
11
+ "train_loss": 0.010893226607249389,
12
+ "train_runtime": 2856.1574,
13
+ "train_samples_per_second": 26.658,
14
+ "train_steps_per_second": 3.333
15
+ }
eval_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9996779906617292,
4
+ "eval_f1": 0.9998205875756896,
5
+ "eval_loss": 0.002155957045033574,
6
+ "eval_precision": 0.9998654346460931,
7
+ "eval_recall": 0.9997757445281665,
8
+ "eval_runtime": 472.1208,
9
+ "eval_samples_per_second": 52.622,
10
+ "eval_steps_per_second": 6.579
11
+ }
runs/Dec21_04-45-05_Phoenix/events.out.tfevents.1703137745.Phoenix.19392.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f13edd75dd9cb3a0429be8f62754ea127eed1c10c54e4eabb3c02cacceaf9fe
3
+ size 1032
test_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.7166781307466625,
4
+ "eval_f1": 0.8124204206436981,
5
+ "eval_loss": 2.0649054050445557,
6
+ "eval_precision": 0.9998169964543063,
7
+ "eval_recall": 0.6841833380294918,
8
+ "eval_runtime": 1515.2798,
9
+ "eval_samples_per_second": 47.012,
10
+ "eval_steps_per_second": 5.877
11
+ }
train_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "train_loss": 0.010893226607249389,
4
+ "train_runtime": 2856.1574,
5
+ "train_samples_per_second": 26.658,
6
+ "train_steps_per_second": 3.333
7
+ }
trainer_state.json ADDED
@@ -0,0 +1,180 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.002155957045033574,
3
+ "best_model_checkpoint": "W:/res/Transformers/my-google/vit-base-patch16-224-in21k\\checkpoint-6346",
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 9519,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.16,
13
+ "learning_rate": 4.7373673705221136e-05,
14
+ "loss": 0.0936,
15
+ "step": 500
16
+ },
17
+ {
18
+ "epoch": 0.32,
19
+ "learning_rate": 4.4747347410442276e-05,
20
+ "loss": 0.019,
21
+ "step": 1000
22
+ },
23
+ {
24
+ "epoch": 0.47,
25
+ "learning_rate": 4.212102111566341e-05,
26
+ "loss": 0.0339,
27
+ "step": 1500
28
+ },
29
+ {
30
+ "epoch": 0.63,
31
+ "learning_rate": 3.949469482088455e-05,
32
+ "loss": 0.0129,
33
+ "step": 2000
34
+ },
35
+ {
36
+ "epoch": 0.79,
37
+ "learning_rate": 3.686836852610568e-05,
38
+ "loss": 0.0154,
39
+ "step": 2500
40
+ },
41
+ {
42
+ "epoch": 0.95,
43
+ "learning_rate": 3.424204223132682e-05,
44
+ "loss": 0.007,
45
+ "step": 3000
46
+ },
47
+ {
48
+ "epoch": 1.0,
49
+ "eval_accuracy": 0.9971824182901304,
50
+ "eval_f1": 0.9984326720702164,
51
+ "eval_loss": 0.010775323025882244,
52
+ "eval_precision": 0.9968702494858267,
53
+ "eval_recall": 1.0,
54
+ "eval_runtime": 484.8551,
55
+ "eval_samples_per_second": 51.24,
56
+ "eval_steps_per_second": 6.406,
57
+ "step": 3173
58
+ },
59
+ {
60
+ "epoch": 1.1,
61
+ "learning_rate": 3.1615715936547956e-05,
62
+ "loss": 0.0068,
63
+ "step": 3500
64
+ },
65
+ {
66
+ "epoch": 1.26,
67
+ "learning_rate": 2.8989389641769092e-05,
68
+ "loss": 0.0047,
69
+ "step": 4000
70
+ },
71
+ {
72
+ "epoch": 1.42,
73
+ "learning_rate": 2.636306334699023e-05,
74
+ "loss": 0.0041,
75
+ "step": 4500
76
+ },
77
+ {
78
+ "epoch": 1.58,
79
+ "learning_rate": 2.373673705221137e-05,
80
+ "loss": 0.0022,
81
+ "step": 5000
82
+ },
83
+ {
84
+ "epoch": 1.73,
85
+ "learning_rate": 2.1110410757432505e-05,
86
+ "loss": 0.0028,
87
+ "step": 5500
88
+ },
89
+ {
90
+ "epoch": 1.89,
91
+ "learning_rate": 1.8484084462653642e-05,
92
+ "loss": 0.0015,
93
+ "step": 6000
94
+ },
95
+ {
96
+ "epoch": 2.0,
97
+ "eval_accuracy": 0.9996779906617292,
98
+ "eval_f1": 0.9998205875756896,
99
+ "eval_loss": 0.002155957045033574,
100
+ "eval_precision": 0.9998654346460931,
101
+ "eval_recall": 0.9997757445281665,
102
+ "eval_runtime": 380.0049,
103
+ "eval_samples_per_second": 65.378,
104
+ "eval_steps_per_second": 8.174,
105
+ "step": 6346
106
+ },
107
+ {
108
+ "epoch": 2.05,
109
+ "learning_rate": 1.585775816787478e-05,
110
+ "loss": 0.0001,
111
+ "step": 6500
112
+ },
113
+ {
114
+ "epoch": 2.21,
115
+ "learning_rate": 1.3231431873095915e-05,
116
+ "loss": 0.0001,
117
+ "step": 7000
118
+ },
119
+ {
120
+ "epoch": 2.36,
121
+ "learning_rate": 1.060510557831705e-05,
122
+ "loss": 0.003,
123
+ "step": 7500
124
+ },
125
+ {
126
+ "epoch": 2.52,
127
+ "learning_rate": 7.978779283538187e-06,
128
+ "loss": 0.0001,
129
+ "step": 8000
130
+ },
131
+ {
132
+ "epoch": 2.68,
133
+ "learning_rate": 5.352452988759324e-06,
134
+ "loss": 0.0,
135
+ "step": 8500
136
+ },
137
+ {
138
+ "epoch": 2.84,
139
+ "learning_rate": 2.7261266939804603e-06,
140
+ "loss": 0.0,
141
+ "step": 9000
142
+ },
143
+ {
144
+ "epoch": 2.99,
145
+ "learning_rate": 9.98003992015968e-08,
146
+ "loss": 0.0,
147
+ "step": 9500
148
+ },
149
+ {
150
+ "epoch": 3.0,
151
+ "eval_accuracy": 0.9996377394944453,
152
+ "eval_f1": 0.9997981836528759,
153
+ "eval_loss": 0.0025263545103371143,
154
+ "eval_precision": 0.9997309296381004,
155
+ "eval_recall": 0.9998654467168999,
156
+ "eval_runtime": 380.4752,
157
+ "eval_samples_per_second": 65.297,
158
+ "eval_steps_per_second": 8.163,
159
+ "step": 9519
160
+ },
161
+ {
162
+ "epoch": 3.0,
163
+ "step": 9519,
164
+ "total_flos": 5.900240089255035e+18,
165
+ "train_loss": 0.010893226607249389,
166
+ "train_runtime": 2856.1574,
167
+ "train_samples_per_second": 26.658,
168
+ "train_steps_per_second": 3.333
169
+ }
170
+ ],
171
+ "logging_steps": 500,
172
+ "max_steps": 9519,
173
+ "num_input_tokens_seen": 0,
174
+ "num_train_epochs": 3,
175
+ "save_steps": 500,
176
+ "total_flos": 5.900240089255035e+18,
177
+ "train_batch_size": 8,
178
+ "trial_name": null,
179
+ "trial_params": null
180
+ }