Hugging Face's logo

mmomm25
/
vit-base-patch16-224-in21k-bridgedefectVIT

Safetensors
vit
Generated from Trainer
Eval Results
Model card Files Community
mmomm25 commited on
Commit
17b9dab
·
verified ·
1 Parent(s): 437ef09

Training in progress, epoch 1

Browse files
Files changed (12) hide show
  1. config.json +38 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +22 -0
  4. tmp-checkpoint-255/config.json +38 -0
  5. tmp-checkpoint-255/model.safetensors +3 -0
  6. tmp-checkpoint-255/optimizer.pt +3 -0
  7. tmp-checkpoint-255/preprocessor_config.json +22 -0
  8. tmp-checkpoint-255/rng_state.pth +3 -0
  9. tmp-checkpoint-255/scheduler.pt +3 -0
  10. tmp-checkpoint-255/trainer_state.json +191 -0
  11. tmp-checkpoint-255/training_args.bin +3 -0
  12. training_args.bin +3 -0
config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "crack",
13
+ "1": "for review",
14
+ "2": "material deterioration",
15
+ "3": "repair",
16
+ "4": "water repair"
17
+ },
18
+ "image_size": 224,
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 3072,
21
+ "label2id": {
22
+ "crack": 0,
23
+ "for review": 1,
24
+ "material deterioration": 2,
25
+ "repair": 3,
26
+ "water repair": 4
27
+ },
28
+ "layer_norm_eps": 1e-12,
29
+ "model_type": "vit",
30
+ "num_attention_heads": 12,
31
+ "num_channels": 3,
32
+ "num_hidden_layers": 12,
33
+ "patch_size": 16,
34
+ "problem_type": "single_label_classification",
35
+ "qkv_bias": true,
36
+ "torch_dtype": "float32",
37
+ "transformers_version": "4.37.2"
38
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f34d330b1301bb075deb7f2c2a2f305d20b3706c152f224a6414538ae253906e
3
+ size 343233204
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
+ }
tmp-checkpoint-255/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "crack",
13
+ "1": "for review",
14
+ "2": "material deterioration",
15
+ "3": "repair",
16
+ "4": "water repair"
17
+ },
18
+ "image_size": 224,
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 3072,
21
+ "label2id": {
22
+ "crack": 0,
23
+ "for review": 1,
24
+ "material deterioration": 2,
25
+ "repair": 3,
26
+ "water repair": 4
27
+ },
28
+ "layer_norm_eps": 1e-12,
29
+ "model_type": "vit",
30
+ "num_attention_heads": 12,
31
+ "num_channels": 3,
32
+ "num_hidden_layers": 12,
33
+ "patch_size": 16,
34
+ "problem_type": "single_label_classification",
35
+ "qkv_bias": true,
36
+ "torch_dtype": "float32",
37
+ "transformers_version": "4.37.2"
38
+ }
tmp-checkpoint-255/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f34d330b1301bb075deb7f2c2a2f305d20b3706c152f224a6414538ae253906e
3
+ size 343233204
tmp-checkpoint-255/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71c7347ba8b2a4df8347196be47645a219585095ba58bb2cc9247d2d00a5192a
3
+ size 686587322
tmp-checkpoint-255/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
+ }
tmp-checkpoint-255/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ebd614adac6a126e3e8292feab79ae3f56a071cbe1c5ecf068e0fe2ad6e4bdb
3
+ size 14244
tmp-checkpoint-255/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb2108d01047144ed28c6a80ce1ff6f4e979992226a38856461eaf7359e21a0c
3
+ size 1064
tmp-checkpoint-255/trainer_state.json ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.158827304840088,
3
+ "best_model_checkpoint": "vit-base-patch16-224-in21k-bridgedefectVIT\\checkpoint-255",
4
+ "epoch": 1.0,
5
+ "eval_steps": 500,
6
+ "global_step": 255,
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.04,
13
+ "learning_rate": 6.535947712418301e-07,
14
+ "loss": 1.6179,
15
+ "step": 10
16
+ },
17
+ {
18
+ "epoch": 0.08,
19
+ "learning_rate": 1.3071895424836602e-06,
20
+ "loss": 1.6223,
21
+ "step": 20
22
+ },
23
+ {
24
+ "epoch": 0.12,
25
+ "learning_rate": 1.96078431372549e-06,
26
+ "loss": 1.6263,
27
+ "step": 30
28
+ },
29
+ {
30
+ "epoch": 0.16,
31
+ "learning_rate": 2.6143790849673204e-06,
32
+ "loss": 1.601,
33
+ "step": 40
34
+ },
35
+ {
36
+ "epoch": 0.2,
37
+ "learning_rate": 3.2679738562091506e-06,
38
+ "loss": 1.6074,
39
+ "step": 50
40
+ },
41
+ {
42
+ "epoch": 0.24,
43
+ "learning_rate": 3.92156862745098e-06,
44
+ "loss": 1.5955,
45
+ "step": 60
46
+ },
47
+ {
48
+ "epoch": 0.27,
49
+ "learning_rate": 4.5751633986928105e-06,
50
+ "loss": 1.5698,
51
+ "step": 70
52
+ },
53
+ {
54
+ "epoch": 0.31,
55
+ "learning_rate": 5.228758169934641e-06,
56
+ "loss": 1.5797,
57
+ "step": 80
58
+ },
59
+ {
60
+ "epoch": 0.35,
61
+ "learning_rate": 5.882352941176471e-06,
62
+ "loss": 1.5531,
63
+ "step": 90
64
+ },
65
+ {
66
+ "epoch": 0.39,
67
+ "learning_rate": 6.535947712418301e-06,
68
+ "loss": 1.5434,
69
+ "step": 100
70
+ },
71
+ {
72
+ "epoch": 0.43,
73
+ "learning_rate": 7.1895424836601305e-06,
74
+ "loss": 1.5433,
75
+ "step": 110
76
+ },
77
+ {
78
+ "epoch": 0.47,
79
+ "learning_rate": 7.84313725490196e-06,
80
+ "loss": 1.5329,
81
+ "step": 120
82
+ },
83
+ {
84
+ "epoch": 0.51,
85
+ "learning_rate": 8.496732026143791e-06,
86
+ "loss": 1.4995,
87
+ "step": 130
88
+ },
89
+ {
90
+ "epoch": 0.55,
91
+ "learning_rate": 9.150326797385621e-06,
92
+ "loss": 1.4875,
93
+ "step": 140
94
+ },
95
+ {
96
+ "epoch": 0.59,
97
+ "learning_rate": 9.803921568627451e-06,
98
+ "loss": 1.4863,
99
+ "step": 150
100
+ },
101
+ {
102
+ "epoch": 0.63,
103
+ "learning_rate": 1.0457516339869281e-05,
104
+ "loss": 1.4209,
105
+ "step": 160
106
+ },
107
+ {
108
+ "epoch": 0.67,
109
+ "learning_rate": 1.1111111111111112e-05,
110
+ "loss": 1.4354,
111
+ "step": 170
112
+ },
113
+ {
114
+ "epoch": 0.71,
115
+ "learning_rate": 1.1764705882352942e-05,
116
+ "loss": 1.3572,
117
+ "step": 180
118
+ },
119
+ {
120
+ "epoch": 0.75,
121
+ "learning_rate": 1.2418300653594772e-05,
122
+ "loss": 1.3714,
123
+ "step": 190
124
+ },
125
+ {
126
+ "epoch": 0.78,
127
+ "learning_rate": 1.3071895424836602e-05,
128
+ "loss": 1.3406,
129
+ "step": 200
130
+ },
131
+ {
132
+ "epoch": 0.82,
133
+ "learning_rate": 1.3725490196078432e-05,
134
+ "loss": 1.2574,
135
+ "step": 210
136
+ },
137
+ {
138
+ "epoch": 0.86,
139
+ "learning_rate": 1.4379084967320261e-05,
140
+ "loss": 1.2965,
141
+ "step": 220
142
+ },
143
+ {
144
+ "epoch": 0.9,
145
+ "learning_rate": 1.5032679738562091e-05,
146
+ "loss": 1.2021,
147
+ "step": 230
148
+ },
149
+ {
150
+ "epoch": 0.94,
151
+ "learning_rate": 1.568627450980392e-05,
152
+ "loss": 1.2072,
153
+ "step": 240
154
+ },
155
+ {
156
+ "epoch": 0.98,
157
+ "learning_rate": 1.6339869281045753e-05,
158
+ "loss": 1.1457,
159
+ "step": 250
160
+ },
161
+ {
162
+ "epoch": 1.0,
163
+ "eval_accuracy": {
164
+ "accuracy": 0.6941176470588235
165
+ },
166
+ "eval_f1": {
167
+ "f1": 0.683792091280419
168
+ },
169
+ "eval_loss": 1.158827304840088,
170
+ "eval_precision": {
171
+ "precision": 0.7377494408375539
172
+ },
173
+ "eval_recall": {
174
+ "recall": 0.6941484490686618
175
+ },
176
+ "eval_runtime": 25.3262,
177
+ "eval_samples_per_second": 20.137,
178
+ "eval_steps_per_second": 10.069,
179
+ "step": 255
180
+ }
181
+ ],
182
+ "logging_steps": 10,
183
+ "max_steps": 7650,
184
+ "num_input_tokens_seen": 0,
185
+ "num_train_epochs": 30,
186
+ "save_steps": 500,
187
+ "total_flos": 1.5808790937526272e+17,
188
+ "train_batch_size": 2,
189
+ "trial_name": null,
190
+ "trial_params": null
191
+ }
tmp-checkpoint-255/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:418b85f35316fefe4bc777808f9dff13bb59b607bf93a8504dc8b5709313507f
3
+ size 4728
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:418b85f35316fefe4bc777808f9dff13bb59b607bf93a8504dc8b5709313507f
3
+ size 4728