bigband commited on
Commit
d9ed84c
·
verified ·
1 Parent(s): 2a9e082

Upload folder using huggingface_hub

Browse files
Files changed (7) hide show
  1. config.json +51 -0
  2. model.safetensors +3 -0
  3. optimizer.pt +3 -0
  4. rng_state.pth +3 -0
  5. scheduler.pt +3 -0
  6. trainer_state.json +186 -0
  7. training_args.bin +3 -0
config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ConvNextForImageClassification"
4
+ ],
5
+ "depths": [
6
+ 3,
7
+ 3,
8
+ 9,
9
+ 3
10
+ ],
11
+ "drop_path_rate": 0.0,
12
+ "dtype": "float32",
13
+ "hidden_act": "gelu",
14
+ "hidden_sizes": [
15
+ 96,
16
+ 192,
17
+ 384,
18
+ 768
19
+ ],
20
+ "id2label": {
21
+ "0": "None",
22
+ "1": "Roadwork"
23
+ },
24
+ "image_size": 224,
25
+ "initializer_range": 0.02,
26
+ "label2id": {
27
+ "None": 0,
28
+ "Roadwork": 1
29
+ },
30
+ "layer_norm_eps": 1e-12,
31
+ "layer_scale_init_value": 1e-06,
32
+ "model_type": "convnext",
33
+ "num_channels": 3,
34
+ "num_stages": 4,
35
+ "out_features": [
36
+ "stage4"
37
+ ],
38
+ "out_indices": [
39
+ 4
40
+ ],
41
+ "patch_size": 4,
42
+ "stage_names": [
43
+ "stem",
44
+ "stage1",
45
+ "stage2",
46
+ "stage3",
47
+ "stage4"
48
+ ],
49
+ "transformers_version": "5.2.0",
50
+ "use_cache": false
51
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0a90bd6da33f4655c84971a0bfdac06cbf830e350bf63c600ce38c2031f72a4
3
+ size 111307920
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14c8f0d4413edf850db27ed237b7b4b61ce1552f78a57fef9447285243163f37
3
+ size 222725963
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:01f9a0f7843a37be87edd23f4e88aa93b38b95cc2c07503eeb1cf2e4632453a2
3
+ size 14645
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff1e843a3330227b0eb7976933fe132e76337bef8f8b6020dc71eddb46e9ce8f
3
+ size 1465
trainer_state.json ADDED
@@ -0,0 +1,186 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 588,
3
+ "best_metric": 0.9346530755096685,
4
+ "best_model_checkpoint": "out_convnext_tiny/checkpoint-588",
5
+ "epoch": 3.0,
6
+ "eval_steps": 500,
7
+ "global_step": 588,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.25510204081632654,
14
+ "grad_norm": 2.420530319213867,
15
+ "learning_rate": 4.829749103942652e-05,
16
+ "loss": 0.5961227035522461,
17
+ "step": 50
18
+ },
19
+ {
20
+ "epoch": 0.5102040816326531,
21
+ "grad_norm": 4.015362739562988,
22
+ "learning_rate": 4.381720430107527e-05,
23
+ "loss": 0.3770874786376953,
24
+ "step": 100
25
+ },
26
+ {
27
+ "epoch": 0.7653061224489796,
28
+ "grad_norm": 1.688482642173767,
29
+ "learning_rate": 3.933691756272402e-05,
30
+ "loss": 0.3206790924072266,
31
+ "step": 150
32
+ },
33
+ {
34
+ "epoch": 1.0,
35
+ "eval_accuracy": 0.9577893820713664,
36
+ "eval_balanced_accuracy": 0.9255876073885376,
37
+ "eval_f1_macro": 0.893579568973049,
38
+ "eval_f1_none": 0.8109161793372319,
39
+ "eval_f1_pos": 0.976242958608866,
40
+ "eval_false_negatives": 70,
41
+ "eval_false_positives": 27,
42
+ "eval_loss": 0.14045794308185577,
43
+ "eval_precision_none": 0.7482014388489209,
44
+ "eval_precision_pos": 0.9866336633663366,
45
+ "eval_recall_none": 0.8851063829787233,
46
+ "eval_recall_pos": 0.9660688317983519,
47
+ "eval_roc_auc": 0.9667206402574231,
48
+ "eval_runtime": 22.5697,
49
+ "eval_samples_per_second": 101.818,
50
+ "eval_steps_per_second": 25.477,
51
+ "eval_true_negatives": 208,
52
+ "eval_true_positives": 1993,
53
+ "step": 196
54
+ },
55
+ {
56
+ "epoch": 1.0204081632653061,
57
+ "grad_norm": 3.1961281299591064,
58
+ "learning_rate": 3.485663082437276e-05,
59
+ "loss": 0.29840837478637694,
60
+ "step": 200
61
+ },
62
+ {
63
+ "epoch": 1.2755102040816326,
64
+ "grad_norm": 0.8268783092498779,
65
+ "learning_rate": 3.0376344086021508e-05,
66
+ "loss": 0.21166513442993165,
67
+ "step": 250
68
+ },
69
+ {
70
+ "epoch": 1.5306122448979593,
71
+ "grad_norm": 1.6696209907531738,
72
+ "learning_rate": 2.589605734767025e-05,
73
+ "loss": 0.22809907913208008,
74
+ "step": 300
75
+ },
76
+ {
77
+ "epoch": 1.7857142857142856,
78
+ "grad_norm": 3.8369956016540527,
79
+ "learning_rate": 2.1415770609318996e-05,
80
+ "loss": 0.21474090576171875,
81
+ "step": 350
82
+ },
83
+ {
84
+ "epoch": 2.0,
85
+ "eval_accuracy": 0.9769364664926022,
86
+ "eval_balanced_accuracy": 0.8966605129897588,
87
+ "eval_f1_macro": 0.9315826696768779,
88
+ "eval_f1_none": 0.8758782201405152,
89
+ "eval_f1_pos": 0.9872871192132406,
90
+ "eval_false_negatives": 5,
91
+ "eval_false_positives": 48,
92
+ "eval_loss": 0.08764547854661942,
93
+ "eval_precision_none": 0.9739583333333334,
94
+ "eval_precision_pos": 0.9772079772079773,
95
+ "eval_recall_none": 0.7957446808510639,
96
+ "eval_recall_pos": 0.9975763451284537,
97
+ "eval_roc_auc": 0.9633398995472406,
98
+ "eval_runtime": 20.3114,
99
+ "eval_samples_per_second": 113.139,
100
+ "eval_steps_per_second": 28.309,
101
+ "eval_true_negatives": 187,
102
+ "eval_true_positives": 2058,
103
+ "step": 392
104
+ },
105
+ {
106
+ "epoch": 2.0408163265306123,
107
+ "grad_norm": 4.356799602508545,
108
+ "learning_rate": 1.693548387096774e-05,
109
+ "loss": 0.2091566276550293,
110
+ "step": 400
111
+ },
112
+ {
113
+ "epoch": 2.295918367346939,
114
+ "grad_norm": 4.339029312133789,
115
+ "learning_rate": 1.2455197132616488e-05,
116
+ "loss": 0.1602233123779297,
117
+ "step": 450
118
+ },
119
+ {
120
+ "epoch": 2.5510204081632653,
121
+ "grad_norm": 3.0306155681610107,
122
+ "learning_rate": 7.974910394265234e-06,
123
+ "loss": 0.1482492733001709,
124
+ "step": 500
125
+ },
126
+ {
127
+ "epoch": 2.806122448979592,
128
+ "grad_norm": 4.387181282043457,
129
+ "learning_rate": 3.4946236559139785e-06,
130
+ "loss": 0.131751708984375,
131
+ "step": 550
132
+ },
133
+ {
134
+ "epoch": 3.0,
135
+ "eval_accuracy": 0.9773716275021758,
136
+ "eval_balanced_accuracy": 0.9100999370881075,
137
+ "eval_f1_macro": 0.9346530755096685,
138
+ "eval_f1_none": 0.8818181818181818,
139
+ "eval_f1_pos": 0.987487969201155,
140
+ "eval_false_negatives": 11,
141
+ "eval_false_positives": 41,
142
+ "eval_loss": 0.08501837402582169,
143
+ "eval_precision_none": 0.9463414634146341,
144
+ "eval_precision_pos": 0.9804108934543717,
145
+ "eval_recall_none": 0.825531914893617,
146
+ "eval_recall_pos": 0.9946679592825982,
147
+ "eval_roc_auc": 0.96461051350543,
148
+ "eval_runtime": 19.4314,
149
+ "eval_samples_per_second": 118.262,
150
+ "eval_steps_per_second": 29.591,
151
+ "eval_true_negatives": 194,
152
+ "eval_true_positives": 2052,
153
+ "step": 588
154
+ }
155
+ ],
156
+ "logging_steps": 50,
157
+ "max_steps": 588,
158
+ "num_input_tokens_seen": 0,
159
+ "num_train_epochs": 3,
160
+ "save_steps": 500,
161
+ "stateful_callbacks": {
162
+ "EarlyStoppingCallback": {
163
+ "args": {
164
+ "early_stopping_patience": 2,
165
+ "early_stopping_threshold": 0.0
166
+ },
167
+ "attributes": {
168
+ "early_stopping_patience_counter": 0
169
+ }
170
+ },
171
+ "TrainerControl": {
172
+ "args": {
173
+ "should_epoch_stop": false,
174
+ "should_evaluate": false,
175
+ "should_log": false,
176
+ "should_save": true,
177
+ "should_training_stop": true
178
+ },
179
+ "attributes": {}
180
+ }
181
+ },
182
+ "total_flos": 4.7121860362571366e+17,
183
+ "train_batch_size": 32,
184
+ "trial_name": null,
185
+ "trial_params": null
186
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a42a9fd8753673064a55701afa6878c7342c476c2822ec03abe89753880ceaf
3
+ size 5201