Hugging Face's logo

dima806
/
dogs_70_breeds_image_detection

Image Classification
Transformers
PyTorch
Safetensors
vit
Inference Endpoints
Model card Files Community
1
dima806 commited on
Commit
99773c7
•
1 Parent(s): 7facce0

Upload folder using huggingface_hub

Browse files
Files changed (12) hide show
  1. checkpoint-7020/config.json +170 -0
  2. checkpoint-7020/optimizer.pt +3 -0
  3. checkpoint-7020/preprocessor_config.json +22 -0
  4. checkpoint-7020/pytorch_model.bin +3 -0
  5. checkpoint-7020/rng_state.pth +3 -0
  6. checkpoint-7020/scheduler.pt +3 -0
  7. checkpoint-7020/trainer_state.json +373 -0
  8. checkpoint-7020/training_args.bin +3 -0
  9. config.json +170 -0
  10. preprocessor_config.json +22 -0
  11. pytorch_model.bin +3 -0
  12. training_args.bin +3 -0
checkpoint-7020/config.json ADDED
@@ -0,0 +1,170 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "Bulldog",
13
+ "1": "Bearded Collie",
14
+ "2": "Boxer",
15
+ "3": "Shiba Inu",
16
+ "4": "African Wild Dog",
17
+ "5": "Mex Hairless",
18
+ "6": "Cocker",
19
+ "7": "Basenji",
20
+ "8": "Lhasa",
21
+ "9": "Blenheim",
22
+ "10": "Rottweiler",
23
+ "11": "Elk Hound",
24
+ "12": "Vizsla",
25
+ "13": "Siberian Husky",
26
+ "14": "Malinois",
27
+ "15": "Pekinese",
28
+ "16": "Saint Bernard",
29
+ "17": "Border Collie",
30
+ "18": "Bull Terrier",
31
+ "19": "Beagle",
32
+ "20": "Airedale",
33
+ "21": "Great Perenees",
34
+ "22": "Shar_Pei",
35
+ "23": "American Spaniel",
36
+ "24": "Corgi",
37
+ "25": "Chow",
38
+ "26": "Afghan",
39
+ "27": "Newfoundland",
40
+ "28": "German Sheperd",
41
+ "29": "Poodle",
42
+ "30": "Pug",
43
+ "31": "Dingo",
44
+ "32": "Bermaise",
45
+ "33": "Pit Bull",
46
+ "34": "Cairn",
47
+ "35": "Labrador",
48
+ "36": "Schnauzer",
49
+ "37": "Chinese Crested",
50
+ "38": "Coyote",
51
+ "39": "Groenendael",
52
+ "40": "Basset",
53
+ "41": "Maltese",
54
+ "42": "Dhole",
55
+ "43": "Bichon Frise",
56
+ "44": "Doberman",
57
+ "45": "Scotch Terrier",
58
+ "46": "Japanese Spaniel",
59
+ "47": "Borzoi",
60
+ "48": "Labradoodle",
61
+ "49": "Boston Terrier",
62
+ "50": "Chihuahua",
63
+ "51": "Bull Mastiff",
64
+ "52": "American Hairless",
65
+ "53": "Great Dane",
66
+ "54": "Pomeranian",
67
+ "55": "Shih-Tzu",
68
+ "56": "Bluetick",
69
+ "57": "Greyhound",
70
+ "58": "Cockapoo",
71
+ "59": "Golden Retriever",
72
+ "60": "Irish Wolfhound",
73
+ "61": "American Spaniel",
74
+ "62": "Clumber",
75
+ "63": "Dalmation",
76
+ "64": "Bloodhound",
77
+ "65": "Komondor",
78
+ "66": "Rhodesian",
79
+ "67": "French Bulldog",
80
+ "68": "Collie",
81
+ "69": "Irish Spaniel",
82
+ "70": "Yorkie"
83
+ },
84
+ "image_size": 224,
85
+ "initializer_range": 0.02,
86
+ "intermediate_size": 3072,
87
+ "label2id": {
88
+ "Afghan": 26,
89
+ "African Wild Dog": 4,
90
+ "Airedale": 20,
91
+ "American Spaniel": 23,
92
+ "American Hairless": 52,
93
+ "American Spaniel": 61,
94
+ "Basenji": 7,
95
+ "Basset": 40,
96
+ "Beagle": 19,
97
+ "Bearded Collie": 1,
98
+ "Bermaise": 32,
99
+ "Bichon Frise": 43,
100
+ "Blenheim": 9,
101
+ "Bloodhound": 64,
102
+ "Bluetick": 56,
103
+ "Border Collie": 17,
104
+ "Borzoi": 47,
105
+ "Boston Terrier": 49,
106
+ "Boxer": 2,
107
+ "Bull Mastiff": 51,
108
+ "Bull Terrier": 18,
109
+ "Bulldog": 0,
110
+ "Cairn": 34,
111
+ "Chihuahua": 50,
112
+ "Chinese Crested": 37,
113
+ "Chow": 25,
114
+ "Clumber": 62,
115
+ "Cockapoo": 58,
116
+ "Cocker": 6,
117
+ "Collie": 68,
118
+ "Corgi": 24,
119
+ "Coyote": 38,
120
+ "Dalmation": 63,
121
+ "Dhole": 42,
122
+ "Dingo": 31,
123
+ "Doberman": 44,
124
+ "Elk Hound": 11,
125
+ "French Bulldog": 67,
126
+ "German Sheperd": 28,
127
+ "Golden Retriever": 59,
128
+ "Great Dane": 53,
129
+ "Great Perenees": 21,
130
+ "Greyhound": 57,
131
+ "Groenendael": 39,
132
+ "Irish Spaniel": 69,
133
+ "Irish Wolfhound": 60,
134
+ "Japanese Spaniel": 46,
135
+ "Komondor": 65,
136
+ "Labradoodle": 48,
137
+ "Labrador": 35,
138
+ "Lhasa": 8,
139
+ "Malinois": 14,
140
+ "Maltese": 41,
141
+ "Mex Hairless": 5,
142
+ "Newfoundland": 27,
143
+ "Pekinese": 15,
144
+ "Pit Bull": 33,
145
+ "Pomeranian": 54,
146
+ "Poodle": 29,
147
+ "Pug": 30,
148
+ "Rhodesian": 66,
149
+ "Rottweiler": 10,
150
+ "Saint Bernard": 16,
151
+ "Schnauzer": 36,
152
+ "Scotch Terrier": 45,
153
+ "Shar_Pei": 22,
154
+ "Shiba Inu": 3,
155
+ "Shih-Tzu": 55,
156
+ "Siberian Husky": 13,
157
+ "Vizsla": 12,
158
+ "Yorkie": 70
159
+ },
160
+ "layer_norm_eps": 1e-12,
161
+ "model_type": "vit",
162
+ "num_attention_heads": 12,
163
+ "num_channels": 3,
164
+ "num_hidden_layers": 12,
165
+ "patch_size": 16,
166
+ "problem_type": "single_label_classification",
167
+ "qkv_bias": true,
168
+ "torch_dtype": "float32",
169
+ "transformers_version": "4.33.0"
170
+ }
checkpoint-7020/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35b0e7b32a0c2398ad3085610cb40e79957dbe5ad8b924b657fa9f8d9d6b3ceb
3
+ size 686992901
checkpoint-7020/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
+ }
checkpoint-7020/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b0ae805f77d683c6577c66deaaad3110c1ceb64e0bd3c30cae6119f49209397
3
+ size 343480941
checkpoint-7020/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b8305a17e0a6b8fdf6ad2f8f84a47841722e6047a125a2cd0ecdeb988c69522
3
+ size 14575
checkpoint-7020/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fd653201d77d8b8009e2d94d355f5a6eb2179a682b9049f43fb10be162ebc0e
3
+ size 627
checkpoint-7020/trainer_state.json ADDED
@@ -0,0 +1,373 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.7137995958328247,
3
+ "best_model_checkpoint": "dogs_70_breeds_image_detection/checkpoint-7020",
4
+ "epoch": 30.0,
5
+ "eval_steps": 500,
6
+ "global_step": 7020,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 1.0,
13
+ "eval_accuracy": 0.5700534759358289,
14
+ "eval_loss": 3.8619625568389893,
15
+ "eval_runtime": 16.943,
16
+ "eval_samples_per_second": 110.37,
17
+ "eval_steps_per_second": 13.811,
18
+ "step": 234
19
+ },
20
+ {
21
+ "epoch": 2.0,
22
+ "eval_accuracy": 0.7903743315508022,
23
+ "eval_loss": 3.442793846130371,
24
+ "eval_runtime": 16.7743,
25
+ "eval_samples_per_second": 111.48,
26
+ "eval_steps_per_second": 13.95,
27
+ "step": 468
28
+ },
29
+ {
30
+ "epoch": 2.14,
31
+ "learning_rate": 9.354375896700143e-06,
32
+ "loss": 3.8066,
33
+ "step": 500
34
+ },
35
+ {
36
+ "epoch": 3.0,
37
+ "eval_accuracy": 0.8385026737967914,
38
+ "eval_loss": 3.106651782989502,
39
+ "eval_runtime": 16.721,
40
+ "eval_samples_per_second": 111.835,
41
+ "eval_steps_per_second": 13.994,
42
+ "step": 702
43
+ },
44
+ {
45
+ "epoch": 4.0,
46
+ "eval_accuracy": 0.8625668449197861,
47
+ "eval_loss": 2.8242247104644775,
48
+ "eval_runtime": 16.6814,
49
+ "eval_samples_per_second": 112.101,
50
+ "eval_steps_per_second": 14.028,
51
+ "step": 936
52
+ },
53
+ {
54
+ "epoch": 4.27,
55
+ "learning_rate": 8.637015781922526e-06,
56
+ "loss": 2.9436,
57
+ "step": 1000
58
+ },
59
+ {
60
+ "epoch": 5.0,
61
+ "eval_accuracy": 0.8748663101604278,
62
+ "eval_loss": 2.583355665206909,
63
+ "eval_runtime": 16.6174,
64
+ "eval_samples_per_second": 112.533,
65
+ "eval_steps_per_second": 14.082,
66
+ "step": 1170
67
+ },
68
+ {
69
+ "epoch": 6.0,
70
+ "eval_accuracy": 0.8903743315508021,
71
+ "eval_loss": 2.3684189319610596,
72
+ "eval_runtime": 16.679,
73
+ "eval_samples_per_second": 112.117,
74
+ "eval_steps_per_second": 14.03,
75
+ "step": 1404
76
+ },
77
+ {
78
+ "epoch": 6.41,
79
+ "learning_rate": 7.919655667144908e-06,
80
+ "loss": 2.3544,
81
+ "step": 1500
82
+ },
83
+ {
84
+ "epoch": 7.0,
85
+ "eval_accuracy": 0.8935828877005347,
86
+ "eval_loss": 2.177172899246216,
87
+ "eval_runtime": 16.7621,
88
+ "eval_samples_per_second": 111.561,
89
+ "eval_steps_per_second": 13.96,
90
+ "step": 1638
91
+ },
92
+ {
93
+ "epoch": 8.0,
94
+ "eval_accuracy": 0.9058823529411765,
95
+ "eval_loss": 2.007451295852661,
96
+ "eval_runtime": 16.6026,
97
+ "eval_samples_per_second": 112.633,
98
+ "eval_steps_per_second": 14.094,
99
+ "step": 1872
100
+ },
101
+ {
102
+ "epoch": 8.55,
103
+ "learning_rate": 7.202295552367289e-06,
104
+ "loss": 1.9032,
105
+ "step": 2000
106
+ },
107
+ {
108
+ "epoch": 9.0,
109
+ "eval_accuracy": 0.902139037433155,
110
+ "eval_loss": 1.8538062572479248,
111
+ "eval_runtime": 16.685,
112
+ "eval_samples_per_second": 112.077,
113
+ "eval_steps_per_second": 14.025,
114
+ "step": 2106
115
+ },
116
+ {
117
+ "epoch": 10.0,
118
+ "eval_accuracy": 0.9058823529411765,
119
+ "eval_loss": 1.7170625925064087,
120
+ "eval_runtime": 16.7001,
121
+ "eval_samples_per_second": 111.975,
122
+ "eval_steps_per_second": 14.012,
123
+ "step": 2340
124
+ },
125
+ {
126
+ "epoch": 10.68,
127
+ "learning_rate": 6.48493543758967e-06,
128
+ "loss": 1.5446,
129
+ "step": 2500
130
+ },
131
+ {
132
+ "epoch": 11.0,
133
+ "eval_accuracy": 0.9074866310160428,
134
+ "eval_loss": 1.5915807485580444,
135
+ "eval_runtime": 16.4868,
136
+ "eval_samples_per_second": 113.424,
137
+ "eval_steps_per_second": 14.193,
138
+ "step": 2574
139
+ },
140
+ {
141
+ "epoch": 12.0,
142
+ "eval_accuracy": 0.9117647058823529,
143
+ "eval_loss": 1.4730502367019653,
144
+ "eval_runtime": 16.6406,
145
+ "eval_samples_per_second": 112.376,
146
+ "eval_steps_per_second": 14.062,
147
+ "step": 2808
148
+ },
149
+ {
150
+ "epoch": 12.82,
151
+ "learning_rate": 5.767575322812053e-06,
152
+ "loss": 1.2629,
153
+ "step": 3000
154
+ },
155
+ {
156
+ "epoch": 13.0,
157
+ "eval_accuracy": 0.9187165775401069,
158
+ "eval_loss": 1.3681436777114868,
159
+ "eval_runtime": 16.7454,
160
+ "eval_samples_per_second": 111.672,
161
+ "eval_steps_per_second": 13.974,
162
+ "step": 3042
163
+ },
164
+ {
165
+ "epoch": 14.0,
166
+ "eval_accuracy": 0.9165775401069519,
167
+ "eval_loss": 1.2735283374786377,
168
+ "eval_runtime": 16.5505,
169
+ "eval_samples_per_second": 112.987,
170
+ "eval_steps_per_second": 14.139,
171
+ "step": 3276
172
+ },
173
+ {
174
+ "epoch": 14.96,
175
+ "learning_rate": 5.050215208034433e-06,
176
+ "loss": 1.0405,
177
+ "step": 3500
178
+ },
179
+ {
180
+ "epoch": 15.0,
181
+ "eval_accuracy": 0.9181818181818182,
182
+ "eval_loss": 1.1897326707839966,
183
+ "eval_runtime": 16.574,
184
+ "eval_samples_per_second": 112.827,
185
+ "eval_steps_per_second": 14.118,
186
+ "step": 3510
187
+ },
188
+ {
189
+ "epoch": 16.0,
190
+ "eval_accuracy": 0.9229946524064171,
191
+ "eval_loss": 1.1123002767562866,
192
+ "eval_runtime": 16.7899,
193
+ "eval_samples_per_second": 111.377,
194
+ "eval_steps_per_second": 13.937,
195
+ "step": 3744
196
+ },
197
+ {
198
+ "epoch": 17.0,
199
+ "eval_accuracy": 0.9262032085561497,
200
+ "eval_loss": 1.0538486242294312,
201
+ "eval_runtime": 16.5745,
202
+ "eval_samples_per_second": 112.824,
203
+ "eval_steps_per_second": 14.118,
204
+ "step": 3978
205
+ },
206
+ {
207
+ "epoch": 17.09,
208
+ "learning_rate": 4.332855093256815e-06,
209
+ "loss": 0.8665,
210
+ "step": 4000
211
+ },
212
+ {
213
+ "epoch": 18.0,
214
+ "eval_accuracy": 0.925668449197861,
215
+ "eval_loss": 0.9949386715888977,
216
+ "eval_runtime": 16.5637,
217
+ "eval_samples_per_second": 112.897,
218
+ "eval_steps_per_second": 14.127,
219
+ "step": 4212
220
+ },
221
+ {
222
+ "epoch": 19.0,
223
+ "eval_accuracy": 0.9262032085561497,
224
+ "eval_loss": 0.9491089582443237,
225
+ "eval_runtime": 16.5976,
226
+ "eval_samples_per_second": 112.667,
227
+ "eval_steps_per_second": 14.098,
228
+ "step": 4446
229
+ },
230
+ {
231
+ "epoch": 19.23,
232
+ "learning_rate": 3.615494978479197e-06,
233
+ "loss": 0.7318,
234
+ "step": 4500
235
+ },
236
+ {
237
+ "epoch": 20.0,
238
+ "eval_accuracy": 0.9315508021390374,
239
+ "eval_loss": 0.8963261246681213,
240
+ "eval_runtime": 16.6332,
241
+ "eval_samples_per_second": 112.426,
242
+ "eval_steps_per_second": 14.068,
243
+ "step": 4680
244
+ },
245
+ {
246
+ "epoch": 21.0,
247
+ "eval_accuracy": 0.9294117647058824,
248
+ "eval_loss": 0.8556962609291077,
249
+ "eval_runtime": 17.2231,
250
+ "eval_samples_per_second": 108.575,
251
+ "eval_steps_per_second": 13.586,
252
+ "step": 4914
253
+ },
254
+ {
255
+ "epoch": 21.37,
256
+ "learning_rate": 2.898134863701578e-06,
257
+ "loss": 0.6322,
258
+ "step": 5000
259
+ },
260
+ {
261
+ "epoch": 22.0,
262
+ "eval_accuracy": 0.9336898395721925,
263
+ "eval_loss": 0.8246510624885559,
264
+ "eval_runtime": 16.769,
265
+ "eval_samples_per_second": 111.515,
266
+ "eval_steps_per_second": 13.954,
267
+ "step": 5148
268
+ },
269
+ {
270
+ "epoch": 23.0,
271
+ "eval_accuracy": 0.932620320855615,
272
+ "eval_loss": 0.8025164604187012,
273
+ "eval_runtime": 16.8215,
274
+ "eval_samples_per_second": 111.167,
275
+ "eval_steps_per_second": 13.911,
276
+ "step": 5382
277
+ },
278
+ {
279
+ "epoch": 23.5,
280
+ "learning_rate": 2.18077474892396e-06,
281
+ "loss": 0.5623,
282
+ "step": 5500
283
+ },
284
+ {
285
+ "epoch": 24.0,
286
+ "eval_accuracy": 0.9299465240641711,
287
+ "eval_loss": 0.778028666973114,
288
+ "eval_runtime": 16.7832,
289
+ "eval_samples_per_second": 111.421,
290
+ "eval_steps_per_second": 13.943,
291
+ "step": 5616
292
+ },
293
+ {
294
+ "epoch": 25.0,
295
+ "eval_accuracy": 0.9347593582887701,
296
+ "eval_loss": 0.7549712657928467,
297
+ "eval_runtime": 16.589,
298
+ "eval_samples_per_second": 112.725,
299
+ "eval_steps_per_second": 14.106,
300
+ "step": 5850
301
+ },
302
+ {
303
+ "epoch": 25.64,
304
+ "learning_rate": 1.4634146341463414e-06,
305
+ "loss": 0.5055,
306
+ "step": 6000
307
+ },
308
+ {
309
+ "epoch": 26.0,
310
+ "eval_accuracy": 0.9336898395721925,
311
+ "eval_loss": 0.7411012649536133,
312
+ "eval_runtime": 16.9543,
313
+ "eval_samples_per_second": 110.297,
314
+ "eval_steps_per_second": 13.802,
315
+ "step": 6084
316
+ },
317
+ {
318
+ "epoch": 27.0,
319
+ "eval_accuracy": 0.9358288770053476,
320
+ "eval_loss": 0.729629397392273,
321
+ "eval_runtime": 16.9047,
322
+ "eval_samples_per_second": 110.62,
323
+ "eval_steps_per_second": 13.842,
324
+ "step": 6318
325
+ },
326
+ {
327
+ "epoch": 27.78,
328
+ "learning_rate": 7.460545193687231e-07,
329
+ "loss": 0.4743,
330
+ "step": 6500
331
+ },
332
+ {
333
+ "epoch": 28.0,
334
+ "eval_accuracy": 0.9331550802139037,
335
+ "eval_loss": 0.7210513949394226,
336
+ "eval_runtime": 16.6036,
337
+ "eval_samples_per_second": 112.626,
338
+ "eval_steps_per_second": 14.093,
339
+ "step": 6552
340
+ },
341
+ {
342
+ "epoch": 29.0,
343
+ "eval_accuracy": 0.9358288770053476,
344
+ "eval_loss": 0.7155081629753113,
345
+ "eval_runtime": 16.7145,
346
+ "eval_samples_per_second": 111.879,
347
+ "eval_steps_per_second": 14.0,
348
+ "step": 6786
349
+ },
350
+ {
351
+ "epoch": 29.91,
352
+ "learning_rate": 2.869440459110474e-08,
353
+ "loss": 0.457,
354
+ "step": 7000
355
+ },
356
+ {
357
+ "epoch": 30.0,
358
+ "eval_accuracy": 0.9352941176470588,
359
+ "eval_loss": 0.7137995958328247,
360
+ "eval_runtime": 16.6482,
361
+ "eval_samples_per_second": 112.325,
362
+ "eval_steps_per_second": 14.056,
363
+ "step": 7020
364
+ }
365
+ ],
366
+ "logging_steps": 500,
367
+ "max_steps": 7020,
368
+ "num_train_epochs": 30,
369
+ "save_steps": 500,
370
+ "total_flos": 1.7390651600583475e+19,
371
+ "trial_name": null,
372
+ "trial_params": null
373
+ }
checkpoint-7020/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d6c6be3416b12aa281c729a3039482650312fc14cf09341ddd80018b101bc3a
3
+ size 4027
config.json ADDED
@@ -0,0 +1,170 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "Bulldog",
13
+ "1": "Bearded Collie",
14
+ "2": "Boxer",
15
+ "3": "Shiba Inu",
16
+ "4": "African Wild Dog",
17
+ "5": "Mex Hairless",
18
+ "6": "Cocker",
19
+ "7": "Basenji",
20
+ "8": "Lhasa",
21
+ "9": "Blenheim",
22
+ "10": "Rottweiler",
23
+ "11": "Elk Hound",
24
+ "12": "Vizsla",
25
+ "13": "Siberian Husky",
26
+ "14": "Malinois",
27
+ "15": "Pekinese",
28
+ "16": "Saint Bernard",
29
+ "17": "Border Collie",
30
+ "18": "Bull Terrier",
31
+ "19": "Beagle",
32
+ "20": "Airedale",
33
+ "21": "Great Perenees",
34
+ "22": "Shar_Pei",
35
+ "23": "American Spaniel",
36
+ "24": "Corgi",
37
+ "25": "Chow",
38
+ "26": "Afghan",
39
+ "27": "Newfoundland",
40
+ "28": "German Sheperd",
41
+ "29": "Poodle",
42
+ "30": "Pug",
43
+ "31": "Dingo",
44
+ "32": "Bermaise",
45
+ "33": "Pit Bull",
46
+ "34": "Cairn",
47
+ "35": "Labrador",
48
+ "36": "Schnauzer",
49
+ "37": "Chinese Crested",
50
+ "38": "Coyote",
51
+ "39": "Groenendael",
52
+ "40": "Basset",
53
+ "41": "Maltese",
54
+ "42": "Dhole",
55
+ "43": "Bichon Frise",
56
+ "44": "Doberman",
57
+ "45": "Scotch Terrier",
58
+ "46": "Japanese Spaniel",
59
+ "47": "Borzoi",
60
+ "48": "Labradoodle",
61
+ "49": "Boston Terrier",
62
+ "50": "Chihuahua",
63
+ "51": "Bull Mastiff",
64
+ "52": "American Hairless",
65
+ "53": "Great Dane",
66
+ "54": "Pomeranian",
67
+ "55": "Shih-Tzu",
68
+ "56": "Bluetick",
69
+ "57": "Greyhound",
70
+ "58": "Cockapoo",
71
+ "59": "Golden Retriever",
72
+ "60": "Irish Wolfhound",
73
+ "61": "American Spaniel",
74
+ "62": "Clumber",
75
+ "63": "Dalmation",
76
+ "64": "Bloodhound",
77
+ "65": "Komondor",
78
+ "66": "Rhodesian",
79
+ "67": "French Bulldog",
80
+ "68": "Collie",
81
+ "69": "Irish Spaniel",
82
+ "70": "Yorkie"
83
+ },
84
+ "image_size": 224,
85
+ "initializer_range": 0.02,
86
+ "intermediate_size": 3072,
87
+ "label2id": {
88
+ "Afghan": 26,
89
+ "African Wild Dog": 4,
90
+ "Airedale": 20,
91
+ "American Spaniel": 23,
92
+ "American Hairless": 52,
93
+ "American Spaniel": 61,
94
+ "Basenji": 7,
95
+ "Basset": 40,
96
+ "Beagle": 19,
97
+ "Bearded Collie": 1,
98
+ "Bermaise": 32,
99
+ "Bichon Frise": 43,
100
+ "Blenheim": 9,
101
+ "Bloodhound": 64,
102
+ "Bluetick": 56,
103
+ "Border Collie": 17,
104
+ "Borzoi": 47,
105
+ "Boston Terrier": 49,
106
+ "Boxer": 2,
107
+ "Bull Mastiff": 51,
108
+ "Bull Terrier": 18,
109
+ "Bulldog": 0,
110
+ "Cairn": 34,
111
+ "Chihuahua": 50,
112
+ "Chinese Crested": 37,
113
+ "Chow": 25,
114
+ "Clumber": 62,
115
+ "Cockapoo": 58,
116
+ "Cocker": 6,
117
+ "Collie": 68,
118
+ "Corgi": 24,
119
+ "Coyote": 38,
120
+ "Dalmation": 63,
121
+ "Dhole": 42,
122
+ "Dingo": 31,
123
+ "Doberman": 44,
124
+ "Elk Hound": 11,
125
+ "French Bulldog": 67,
126
+ "German Sheperd": 28,
127
+ "Golden Retriever": 59,
128
+ "Great Dane": 53,
129
+ "Great Perenees": 21,
130
+ "Greyhound": 57,
131
+ "Groenendael": 39,
132
+ "Irish Spaniel": 69,
133
+ "Irish Wolfhound": 60,
134
+ "Japanese Spaniel": 46,
135
+ "Komondor": 65,
136
+ "Labradoodle": 48,
137
+ "Labrador": 35,
138
+ "Lhasa": 8,
139
+ "Malinois": 14,
140
+ "Maltese": 41,
141
+ "Mex Hairless": 5,
142
+ "Newfoundland": 27,
143
+ "Pekinese": 15,
144
+ "Pit Bull": 33,
145
+ "Pomeranian": 54,
146
+ "Poodle": 29,
147
+ "Pug": 30,
148
+ "Rhodesian": 66,
149
+ "Rottweiler": 10,
150
+ "Saint Bernard": 16,
151
+ "Schnauzer": 36,
152
+ "Scotch Terrier": 45,
153
+ "Shar_Pei": 22,
154
+ "Shiba Inu": 3,
155
+ "Shih-Tzu": 55,
156
+ "Siberian Husky": 13,
157
+ "Vizsla": 12,
158
+ "Yorkie": 70
159
+ },
160
+ "layer_norm_eps": 1e-12,
161
+ "model_type": "vit",
162
+ "num_attention_heads": 12,
163
+ "num_channels": 3,
164
+ "num_hidden_layers": 12,
165
+ "patch_size": 16,
166
+ "problem_type": "single_label_classification",
167
+ "qkv_bias": true,
168
+ "torch_dtype": "float32",
169
+ "transformers_version": "4.33.0"
170
+ }
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
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b0ae805f77d683c6577c66deaaad3110c1ceb64e0bd3c30cae6119f49209397
3
+ size 343480941
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d6c6be3416b12aa281c729a3039482650312fc14cf09341ddd80018b101bc3a
3
+ size 4027