yuanhuaisen commited on
Commit
6c7e4aa
1 Parent(s): 868f3ff

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - image-classification
6
+ widget:
7
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
8
+ example_title: Tiger
9
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg
10
+ example_title: Teapot
11
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
12
+ example_title: Palace
13
+ datasets:
14
+ - yuanhuaisen/autotrain-data-autotrain-koz62-88avl
15
+ ---
16
+
17
+ # Model Trained Using AutoTrain
18
+
19
+ - Problem type: Image Classification
20
+
21
+ ## Validation Metricsg
22
+ loss: 0.5237402319908142
23
+
24
+ f1: 0.9333333333333333
25
+
26
+ precision: 0.875
27
+
28
+ recall: 1.0
29
+
30
+ auc: 0.9805194805194805
31
+
32
+ accuracy: 0.92
checkpoint-39/config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-large-patch16-224",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "ViTForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "encoder_stride": 16,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 1024,
12
+ "id2label": {
13
+ "0": "11covered_with_a_quilt_and_only_the_head_exposed",
14
+ "1": "12covered_with_a_quilt_and_exposed_other_parts_of_the_body"
15
+ },
16
+ "image_size": 224,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 4096,
19
+ "label2id": {
20
+ "11covered_with_a_quilt_and_only_the_head_exposed": 0,
21
+ "12covered_with_a_quilt_and_exposed_other_parts_of_the_body": 1
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "model_type": "vit",
25
+ "num_attention_heads": 16,
26
+ "num_channels": 3,
27
+ "num_hidden_layers": 24,
28
+ "patch_size": 16,
29
+ "problem_type": "single_label_classification",
30
+ "qkv_bias": true,
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.36.1"
33
+ }
checkpoint-39/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:106136c7cfd0ebc65929c9fcb501781c93f9e64f204255c4fbfac1ffd11f90a8
3
+ size 1213261264
checkpoint-39/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a464a5e2a20111ed1fb61deeb8fa9d796a57a60c123dd8ff3b730234e6b5acf3
3
+ size 2426757546
checkpoint-39/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a96b8d53c05ff0a724e2ac2e60e2ac2dc31ec2049fb3c445d350d44628801f1
3
+ size 14244
checkpoint-39/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13f7de4f4401721240bbc28b20d943d6d5cf41d9d99f74e091f21335a465a41a
3
+ size 1064
checkpoint-39/trainer_state.json ADDED
@@ -0,0 +1,294 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.5237402319908142,
3
+ "best_model_checkpoint": "/tmp/model/checkpoint-39",
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 39,
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.08,
13
+ "learning_rate": 1.25e-05,
14
+ "loss": 0.6934,
15
+ "step": 1
16
+ },
17
+ {
18
+ "epoch": 0.15,
19
+ "learning_rate": 2.5e-05,
20
+ "loss": 0.6934,
21
+ "step": 2
22
+ },
23
+ {
24
+ "epoch": 0.23,
25
+ "learning_rate": 3.7500000000000003e-05,
26
+ "loss": 0.6929,
27
+ "step": 3
28
+ },
29
+ {
30
+ "epoch": 0.31,
31
+ "learning_rate": 5e-05,
32
+ "loss": 0.6935,
33
+ "step": 4
34
+ },
35
+ {
36
+ "epoch": 0.38,
37
+ "learning_rate": 5e-05,
38
+ "loss": 0.6815,
39
+ "step": 5
40
+ },
41
+ {
42
+ "epoch": 0.46,
43
+ "learning_rate": 4.8571428571428576e-05,
44
+ "loss": 0.6877,
45
+ "step": 6
46
+ },
47
+ {
48
+ "epoch": 0.54,
49
+ "learning_rate": 4.714285714285714e-05,
50
+ "loss": 0.6945,
51
+ "step": 7
52
+ },
53
+ {
54
+ "epoch": 0.62,
55
+ "learning_rate": 4.5714285714285716e-05,
56
+ "loss": 0.6888,
57
+ "step": 8
58
+ },
59
+ {
60
+ "epoch": 0.69,
61
+ "learning_rate": 4.428571428571428e-05,
62
+ "loss": 0.6822,
63
+ "step": 9
64
+ },
65
+ {
66
+ "epoch": 0.77,
67
+ "learning_rate": 4.2857142857142856e-05,
68
+ "loss": 0.6884,
69
+ "step": 10
70
+ },
71
+ {
72
+ "epoch": 0.85,
73
+ "learning_rate": 4.1428571428571437e-05,
74
+ "loss": 0.691,
75
+ "step": 11
76
+ },
77
+ {
78
+ "epoch": 0.92,
79
+ "learning_rate": 4e-05,
80
+ "loss": 0.6899,
81
+ "step": 12
82
+ },
83
+ {
84
+ "epoch": 1.0,
85
+ "learning_rate": 3.857142857142858e-05,
86
+ "loss": 0.6825,
87
+ "step": 13
88
+ },
89
+ {
90
+ "epoch": 1.0,
91
+ "eval_accuracy": 0.76,
92
+ "eval_auc": 1.0,
93
+ "eval_f1": 0.8235294117647058,
94
+ "eval_loss": 0.6659374833106995,
95
+ "eval_precision": 0.7,
96
+ "eval_recall": 1.0,
97
+ "eval_runtime": 1.336,
98
+ "eval_samples_per_second": 18.712,
99
+ "eval_steps_per_second": 1.497,
100
+ "step": 13
101
+ },
102
+ {
103
+ "epoch": 1.08,
104
+ "learning_rate": 3.7142857142857143e-05,
105
+ "loss": 0.6623,
106
+ "step": 14
107
+ },
108
+ {
109
+ "epoch": 1.15,
110
+ "learning_rate": 3.571428571428572e-05,
111
+ "loss": 0.656,
112
+ "step": 15
113
+ },
114
+ {
115
+ "epoch": 1.23,
116
+ "learning_rate": 3.428571428571429e-05,
117
+ "loss": 0.6461,
118
+ "step": 16
119
+ },
120
+ {
121
+ "epoch": 1.31,
122
+ "learning_rate": 3.285714285714286e-05,
123
+ "loss": 0.6395,
124
+ "step": 17
125
+ },
126
+ {
127
+ "epoch": 1.38,
128
+ "learning_rate": 3.142857142857143e-05,
129
+ "loss": 0.6732,
130
+ "step": 18
131
+ },
132
+ {
133
+ "epoch": 1.46,
134
+ "learning_rate": 3e-05,
135
+ "loss": 0.644,
136
+ "step": 19
137
+ },
138
+ {
139
+ "epoch": 1.54,
140
+ "learning_rate": 2.857142857142857e-05,
141
+ "loss": 0.6302,
142
+ "step": 20
143
+ },
144
+ {
145
+ "epoch": 1.62,
146
+ "learning_rate": 2.714285714285714e-05,
147
+ "loss": 0.6435,
148
+ "step": 21
149
+ },
150
+ {
151
+ "epoch": 1.69,
152
+ "learning_rate": 2.5714285714285714e-05,
153
+ "loss": 0.6151,
154
+ "step": 22
155
+ },
156
+ {
157
+ "epoch": 1.77,
158
+ "learning_rate": 2.4285714285714288e-05,
159
+ "loss": 0.6093,
160
+ "step": 23
161
+ },
162
+ {
163
+ "epoch": 1.85,
164
+ "learning_rate": 2.2857142857142858e-05,
165
+ "loss": 0.5877,
166
+ "step": 24
167
+ },
168
+ {
169
+ "epoch": 1.92,
170
+ "learning_rate": 2.1428571428571428e-05,
171
+ "loss": 0.6379,
172
+ "step": 25
173
+ },
174
+ {
175
+ "epoch": 2.0,
176
+ "learning_rate": 2e-05,
177
+ "loss": 0.6169,
178
+ "step": 26
179
+ },
180
+ {
181
+ "epoch": 2.0,
182
+ "eval_accuracy": 0.84,
183
+ "eval_auc": 1.0,
184
+ "eval_f1": 0.8750000000000001,
185
+ "eval_loss": 0.5767577886581421,
186
+ "eval_precision": 0.7777777777777778,
187
+ "eval_recall": 1.0,
188
+ "eval_runtime": 1.3558,
189
+ "eval_samples_per_second": 18.439,
190
+ "eval_steps_per_second": 1.475,
191
+ "step": 26
192
+ },
193
+ {
194
+ "epoch": 2.08,
195
+ "learning_rate": 1.8571428571428572e-05,
196
+ "loss": 0.5771,
197
+ "step": 27
198
+ },
199
+ {
200
+ "epoch": 2.15,
201
+ "learning_rate": 1.7142857142857145e-05,
202
+ "loss": 0.5577,
203
+ "step": 28
204
+ },
205
+ {
206
+ "epoch": 2.23,
207
+ "learning_rate": 1.5714285714285715e-05,
208
+ "loss": 0.5667,
209
+ "step": 29
210
+ },
211
+ {
212
+ "epoch": 2.31,
213
+ "learning_rate": 1.4285714285714285e-05,
214
+ "loss": 0.5142,
215
+ "step": 30
216
+ },
217
+ {
218
+ "epoch": 2.38,
219
+ "learning_rate": 1.2857142857142857e-05,
220
+ "loss": 0.5788,
221
+ "step": 31
222
+ },
223
+ {
224
+ "epoch": 2.46,
225
+ "learning_rate": 1.1428571428571429e-05,
226
+ "loss": 0.5735,
227
+ "step": 32
228
+ },
229
+ {
230
+ "epoch": 2.54,
231
+ "learning_rate": 1e-05,
232
+ "loss": 0.5225,
233
+ "step": 33
234
+ },
235
+ {
236
+ "epoch": 2.62,
237
+ "learning_rate": 8.571428571428573e-06,
238
+ "loss": 0.5565,
239
+ "step": 34
240
+ },
241
+ {
242
+ "epoch": 2.69,
243
+ "learning_rate": 7.142857142857143e-06,
244
+ "loss": 0.5518,
245
+ "step": 35
246
+ },
247
+ {
248
+ "epoch": 2.77,
249
+ "learning_rate": 5.7142857142857145e-06,
250
+ "loss": 0.5239,
251
+ "step": 36
252
+ },
253
+ {
254
+ "epoch": 2.85,
255
+ "learning_rate": 4.285714285714286e-06,
256
+ "loss": 0.4709,
257
+ "step": 37
258
+ },
259
+ {
260
+ "epoch": 2.92,
261
+ "learning_rate": 2.8571428571428573e-06,
262
+ "loss": 0.5033,
263
+ "step": 38
264
+ },
265
+ {
266
+ "epoch": 3.0,
267
+ "learning_rate": 1.4285714285714286e-06,
268
+ "loss": 0.5934,
269
+ "step": 39
270
+ },
271
+ {
272
+ "epoch": 3.0,
273
+ "eval_accuracy": 0.92,
274
+ "eval_auc": 0.9805194805194805,
275
+ "eval_f1": 0.9333333333333333,
276
+ "eval_loss": 0.5237402319908142,
277
+ "eval_precision": 0.875,
278
+ "eval_recall": 1.0,
279
+ "eval_runtime": 1.3337,
280
+ "eval_samples_per_second": 18.745,
281
+ "eval_steps_per_second": 1.5,
282
+ "step": 39
283
+ }
284
+ ],
285
+ "logging_steps": 1,
286
+ "max_steps": 39,
287
+ "num_input_tokens_seen": 0,
288
+ "num_train_epochs": 3,
289
+ "save_steps": 500,
290
+ "total_flos": 8.21802539593728e+16,
291
+ "train_batch_size": 8,
292
+ "trial_name": null,
293
+ "trial_params": null
294
+ }
checkpoint-39/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9f848fadcf391ee04e2e420bba7ba8803ccd3ac5fbec726df20ada7cb59d816
3
+ size 4728
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-large-patch16-224",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "ViTForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "encoder_stride": 16,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 1024,
12
+ "id2label": {
13
+ "0": "11covered_with_a_quilt_and_only_the_head_exposed",
14
+ "1": "12covered_with_a_quilt_and_exposed_other_parts_of_the_body"
15
+ },
16
+ "image_size": 224,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 4096,
19
+ "label2id": {
20
+ "11covered_with_a_quilt_and_only_the_head_exposed": 0,
21
+ "12covered_with_a_quilt_and_exposed_other_parts_of_the_body": 1
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "model_type": "vit",
25
+ "num_attention_heads": 16,
26
+ "num_channels": 3,
27
+ "num_hidden_layers": 24,
28
+ "patch_size": 16,
29
+ "problem_type": "single_label_classification",
30
+ "qkv_bias": true,
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.36.1"
33
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:106136c7cfd0ebc65929c9fcb501781c93f9e64f204255c4fbfac1ffd11f90a8
3
+ size 1213261264
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9f848fadcf391ee04e2e420bba7ba8803ccd3ac5fbec726df20ada7cb59d816
3
+ size 4728
training_params.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "yuanhuaisen/autotrain-data-autotrain-koz62-88avl",
3
+ "model": "google/vit-large-patch16-224",
4
+ "username": "yuanhuaisen",
5
+ "lr": 5e-05,
6
+ "epochs": 3,
7
+ "batch_size": 8,
8
+ "warmup_ratio": 0.1,
9
+ "gradient_accumulation": 1,
10
+ "optimizer": "adamw_torch",
11
+ "scheduler": "linear",
12
+ "weight_decay": 0.0,
13
+ "max_grad_norm": 1.0,
14
+ "seed": 42,
15
+ "train_split": "train",
16
+ "valid_split": "validation",
17
+ "logging_steps": -1,
18
+ "project_name": "/tmp/model",
19
+ "auto_find_batch_size": false,
20
+ "mixed_precision": "fp16",
21
+ "save_total_limit": 1,
22
+ "save_strategy": "epoch",
23
+ "push_to_hub": true,
24
+ "repo_id": "yuanhuaisen/autotrain-koz62-88avl",
25
+ "evaluation_strategy": "epoch",
26
+ "image_column": "autotrain_image",
27
+ "target_column": "autotrain_label",
28
+ "log": "none"
29
+ }