PuxAI commited on
Commit
9694891
·
verified ·
1 Parent(s): ec02dd6

Upload batch 24/24: vsfc/mBERT/mha_attention/seed_44 ... vsfc/mBERT/multi_branch_average/seed_44

Browse files
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mha_attention/seed_44/run_command.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ /usr/local/bin/python /workspace/run_glue_MHA_gated.py --model_name_or_path bert-base-multilingual-cased --dataset_name tmnam20/ViGLUE --task_name vsfc --max_seq_length 128 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 25 --report_to none --num_train_epochs 3 --learning_rate 3e-05 --per_device_train_batch_size 48 --per_device_eval_batch_size 96 --gradient_accumulation_steps 1 --dataloader_num_workers 1 --seed 44 --output_dir result_ablation_mbert/vsfc/mBERT/mha_attention/seed_44 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy mha_attention --n_heads 4 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_42/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9412507896399241,
4
+ "eval_loss": 0.21253333985805511,
5
+ "eval_runtime": 6.741,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 234.833,
8
+ "eval_steps_per_second": 2.522,
9
+ "total_flos": 1056957384788088.0,
10
+ "train_loss": 0.20452363604115975,
11
+ "train_runtime": 98.7034,
12
+ "train_samples": 11426,
13
+ "train_samples_per_second": 347.283,
14
+ "train_steps_per_second": 7.264
15
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_42/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9412507896399241,
4
+ "eval_loss": 0.21253333985805511,
5
+ "eval_runtime": 6.741,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 234.833,
8
+ "eval_steps_per_second": 2.522
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_42/run_command.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ /usr/local/bin/python /workspace/run_glue_MHA_gated.py --model_name_or_path bert-base-multilingual-cased --dataset_name tmnam20/ViGLUE --task_name vsfc --max_seq_length 128 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 25 --report_to none --num_train_epochs 3 --learning_rate 3e-05 --per_device_train_batch_size 48 --per_device_eval_batch_size 96 --gradient_accumulation_steps 1 --dataloader_num_workers 1 --seed 42 --output_dir result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_42 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy multi_branch_average --n_heads 4 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_42/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 1056957384788088.0,
4
+ "train_loss": 0.20452363604115975,
5
+ "train_runtime": 98.7034,
6
+ "train_samples": 11426,
7
+ "train_samples_per_second": 347.283,
8
+ "train_steps_per_second": 7.264
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_42/trainer_state.json ADDED
@@ -0,0 +1,265 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 717,
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.10460251046025104,
13
+ "grad_norm": 14.201335906982422,
14
+ "learning_rate": 2.8953974895397488e-05,
15
+ "loss": 0.5405,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.20920502092050208,
20
+ "grad_norm": 5.348233699798584,
21
+ "learning_rate": 2.7907949790794982e-05,
22
+ "loss": 0.3294,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.3138075313807531,
27
+ "grad_norm": 7.623049259185791,
28
+ "learning_rate": 2.686192468619247e-05,
29
+ "loss": 0.3422,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 0.41841004184100417,
34
+ "grad_norm": 5.9439311027526855,
35
+ "learning_rate": 2.5815899581589957e-05,
36
+ "loss": 0.2934,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 0.5230125523012552,
41
+ "grad_norm": 4.006845951080322,
42
+ "learning_rate": 2.4769874476987448e-05,
43
+ "loss": 0.2685,
44
+ "step": 125
45
+ },
46
+ {
47
+ "epoch": 0.6276150627615062,
48
+ "grad_norm": 5.4574055671691895,
49
+ "learning_rate": 2.372384937238494e-05,
50
+ "loss": 0.263,
51
+ "step": 150
52
+ },
53
+ {
54
+ "epoch": 0.7322175732217573,
55
+ "grad_norm": 6.177022457122803,
56
+ "learning_rate": 2.2677824267782426e-05,
57
+ "loss": 0.2551,
58
+ "step": 175
59
+ },
60
+ {
61
+ "epoch": 0.8368200836820083,
62
+ "grad_norm": 5.705365180969238,
63
+ "learning_rate": 2.1631799163179917e-05,
64
+ "loss": 0.2929,
65
+ "step": 200
66
+ },
67
+ {
68
+ "epoch": 0.9414225941422594,
69
+ "grad_norm": 5.65452766418457,
70
+ "learning_rate": 2.0585774058577408e-05,
71
+ "loss": 0.2615,
72
+ "step": 225
73
+ },
74
+ {
75
+ "epoch": 1.0,
76
+ "eval_accuracy": 0.9311433986102338,
77
+ "eval_loss": 0.22236785292625427,
78
+ "eval_runtime": 5.7488,
79
+ "eval_samples_per_second": 275.361,
80
+ "eval_steps_per_second": 2.957,
81
+ "step": 239
82
+ },
83
+ {
84
+ "epoch": 1.0460251046025104,
85
+ "grad_norm": 4.936914920806885,
86
+ "learning_rate": 1.9539748953974895e-05,
87
+ "loss": 0.1962,
88
+ "step": 250
89
+ },
90
+ {
91
+ "epoch": 1.1506276150627615,
92
+ "grad_norm": 8.016222953796387,
93
+ "learning_rate": 1.8493723849372386e-05,
94
+ "loss": 0.192,
95
+ "step": 275
96
+ },
97
+ {
98
+ "epoch": 1.2552301255230125,
99
+ "grad_norm": 3.698970079421997,
100
+ "learning_rate": 1.7447698744769877e-05,
101
+ "loss": 0.1453,
102
+ "step": 300
103
+ },
104
+ {
105
+ "epoch": 1.3598326359832635,
106
+ "grad_norm": 3.36641001701355,
107
+ "learning_rate": 1.6401673640167365e-05,
108
+ "loss": 0.2161,
109
+ "step": 325
110
+ },
111
+ {
112
+ "epoch": 1.4644351464435146,
113
+ "grad_norm": 2.944875955581665,
114
+ "learning_rate": 1.5355648535564852e-05,
115
+ "loss": 0.2049,
116
+ "step": 350
117
+ },
118
+ {
119
+ "epoch": 1.5690376569037658,
120
+ "grad_norm": 6.417823314666748,
121
+ "learning_rate": 1.4309623430962344e-05,
122
+ "loss": 0.192,
123
+ "step": 375
124
+ },
125
+ {
126
+ "epoch": 1.6736401673640167,
127
+ "grad_norm": 3.3958539962768555,
128
+ "learning_rate": 1.3263598326359832e-05,
129
+ "loss": 0.1681,
130
+ "step": 400
131
+ },
132
+ {
133
+ "epoch": 1.778242677824268,
134
+ "grad_norm": 11.595094680786133,
135
+ "learning_rate": 1.2217573221757323e-05,
136
+ "loss": 0.1696,
137
+ "step": 425
138
+ },
139
+ {
140
+ "epoch": 1.8828451882845187,
141
+ "grad_norm": 6.112818241119385,
142
+ "learning_rate": 1.1171548117154812e-05,
143
+ "loss": 0.1422,
144
+ "step": 450
145
+ },
146
+ {
147
+ "epoch": 1.98744769874477,
148
+ "grad_norm": 8.460183143615723,
149
+ "learning_rate": 1.0125523012552301e-05,
150
+ "loss": 0.1938,
151
+ "step": 475
152
+ },
153
+ {
154
+ "epoch": 2.0,
155
+ "eval_accuracy": 0.9374605180037903,
156
+ "eval_loss": 0.20332568883895874,
157
+ "eval_runtime": 4.7706,
158
+ "eval_samples_per_second": 331.821,
159
+ "eval_steps_per_second": 3.563,
160
+ "step": 478
161
+ },
162
+ {
163
+ "epoch": 2.092050209205021,
164
+ "grad_norm": 6.223247051239014,
165
+ "learning_rate": 9.079497907949792e-06,
166
+ "loss": 0.1168,
167
+ "step": 500
168
+ },
169
+ {
170
+ "epoch": 2.196652719665272,
171
+ "grad_norm": 6.7170090675354,
172
+ "learning_rate": 8.03347280334728e-06,
173
+ "loss": 0.1232,
174
+ "step": 525
175
+ },
176
+ {
177
+ "epoch": 2.301255230125523,
178
+ "grad_norm": 7.830327033996582,
179
+ "learning_rate": 6.98744769874477e-06,
180
+ "loss": 0.155,
181
+ "step": 550
182
+ },
183
+ {
184
+ "epoch": 2.405857740585774,
185
+ "grad_norm": 8.447519302368164,
186
+ "learning_rate": 5.941422594142259e-06,
187
+ "loss": 0.1328,
188
+ "step": 575
189
+ },
190
+ {
191
+ "epoch": 2.510460251046025,
192
+ "grad_norm": 5.235635280609131,
193
+ "learning_rate": 4.895397489539749e-06,
194
+ "loss": 0.1044,
195
+ "step": 600
196
+ },
197
+ {
198
+ "epoch": 2.6150627615062763,
199
+ "grad_norm": 6.357935905456543,
200
+ "learning_rate": 3.849372384937238e-06,
201
+ "loss": 0.1431,
202
+ "step": 625
203
+ },
204
+ {
205
+ "epoch": 2.719665271966527,
206
+ "grad_norm": 4.409952640533447,
207
+ "learning_rate": 2.803347280334728e-06,
208
+ "loss": 0.1192,
209
+ "step": 650
210
+ },
211
+ {
212
+ "epoch": 2.8242677824267783,
213
+ "grad_norm": 2.4843714237213135,
214
+ "learning_rate": 1.7573221757322176e-06,
215
+ "loss": 0.1143,
216
+ "step": 675
217
+ },
218
+ {
219
+ "epoch": 2.928870292887029,
220
+ "grad_norm": 5.9728851318359375,
221
+ "learning_rate": 7.112970711297071e-07,
222
+ "loss": 0.111,
223
+ "step": 700
224
+ },
225
+ {
226
+ "epoch": 3.0,
227
+ "eval_accuracy": 0.9412507896399241,
228
+ "eval_loss": 0.21253333985805511,
229
+ "eval_runtime": 5.2439,
230
+ "eval_samples_per_second": 301.876,
231
+ "eval_steps_per_second": 3.242,
232
+ "step": 717
233
+ },
234
+ {
235
+ "epoch": 3.0,
236
+ "step": 717,
237
+ "total_flos": 1056957384788088.0,
238
+ "train_loss": 0.20452363604115975,
239
+ "train_runtime": 98.7034,
240
+ "train_samples_per_second": 347.283,
241
+ "train_steps_per_second": 7.264
242
+ }
243
+ ],
244
+ "logging_steps": 25,
245
+ "max_steps": 717,
246
+ "num_input_tokens_seen": 0,
247
+ "num_train_epochs": 3,
248
+ "save_steps": 500,
249
+ "stateful_callbacks": {
250
+ "TrainerControl": {
251
+ "args": {
252
+ "should_epoch_stop": false,
253
+ "should_evaluate": false,
254
+ "should_log": false,
255
+ "should_save": false,
256
+ "should_training_stop": false
257
+ },
258
+ "attributes": {}
259
+ }
260
+ },
261
+ "total_flos": 1056957384788088.0,
262
+ "train_batch_size": 48,
263
+ "trial_name": null,
264
+ "trial_params": null
265
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_43/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9330385344283006,
4
+ "eval_loss": 0.22735266387462616,
5
+ "eval_runtime": 4.952,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 319.67,
8
+ "eval_steps_per_second": 3.433,
9
+ "total_flos": 1055495423686536.0,
10
+ "train_loss": 0.20476110576752007,
11
+ "train_runtime": 98.8837,
12
+ "train_samples": 11426,
13
+ "train_samples_per_second": 346.65,
14
+ "train_steps_per_second": 7.251
15
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_43/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9330385344283006,
4
+ "eval_loss": 0.22735266387462616,
5
+ "eval_runtime": 4.952,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 319.67,
8
+ "eval_steps_per_second": 3.433
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_43/run_command.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ /usr/local/bin/python /workspace/run_glue_MHA_gated.py --model_name_or_path bert-base-multilingual-cased --dataset_name tmnam20/ViGLUE --task_name vsfc --max_seq_length 128 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 25 --report_to none --num_train_epochs 3 --learning_rate 3e-05 --per_device_train_batch_size 48 --per_device_eval_batch_size 96 --gradient_accumulation_steps 1 --dataloader_num_workers 1 --seed 43 --output_dir result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_43 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy multi_branch_average --n_heads 4 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_43/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 1055495423686536.0,
4
+ "train_loss": 0.20476110576752007,
5
+ "train_runtime": 98.8837,
6
+ "train_samples": 11426,
7
+ "train_samples_per_second": 346.65,
8
+ "train_steps_per_second": 7.251
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_43/trainer_state.json ADDED
@@ -0,0 +1,265 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 717,
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.10460251046025104,
13
+ "grad_norm": 5.623245716094971,
14
+ "learning_rate": 2.8953974895397488e-05,
15
+ "loss": 0.5208,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.20920502092050208,
20
+ "grad_norm": 7.216926097869873,
21
+ "learning_rate": 2.7907949790794982e-05,
22
+ "loss": 0.3094,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.3138075313807531,
27
+ "grad_norm": 7.644581317901611,
28
+ "learning_rate": 2.686192468619247e-05,
29
+ "loss": 0.2967,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 0.41841004184100417,
34
+ "grad_norm": 4.29410982131958,
35
+ "learning_rate": 2.5815899581589957e-05,
36
+ "loss": 0.2768,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 0.5230125523012552,
41
+ "grad_norm": 5.988767147064209,
42
+ "learning_rate": 2.4769874476987448e-05,
43
+ "loss": 0.3272,
44
+ "step": 125
45
+ },
46
+ {
47
+ "epoch": 0.6276150627615062,
48
+ "grad_norm": 6.748810768127441,
49
+ "learning_rate": 2.372384937238494e-05,
50
+ "loss": 0.2716,
51
+ "step": 150
52
+ },
53
+ {
54
+ "epoch": 0.7322175732217573,
55
+ "grad_norm": 4.940288066864014,
56
+ "learning_rate": 2.2677824267782426e-05,
57
+ "loss": 0.2694,
58
+ "step": 175
59
+ },
60
+ {
61
+ "epoch": 0.8368200836820083,
62
+ "grad_norm": 8.23596477508545,
63
+ "learning_rate": 2.1631799163179917e-05,
64
+ "loss": 0.2408,
65
+ "step": 200
66
+ },
67
+ {
68
+ "epoch": 0.9414225941422594,
69
+ "grad_norm": 5.696824073791504,
70
+ "learning_rate": 2.0585774058577408e-05,
71
+ "loss": 0.2343,
72
+ "step": 225
73
+ },
74
+ {
75
+ "epoch": 1.0,
76
+ "eval_accuracy": 0.9222994314592545,
77
+ "eval_loss": 0.24964654445648193,
78
+ "eval_runtime": 6.5874,
79
+ "eval_samples_per_second": 240.306,
80
+ "eval_steps_per_second": 2.581,
81
+ "step": 239
82
+ },
83
+ {
84
+ "epoch": 1.0460251046025104,
85
+ "grad_norm": 3.607553720474243,
86
+ "learning_rate": 1.9539748953974895e-05,
87
+ "loss": 0.2256,
88
+ "step": 250
89
+ },
90
+ {
91
+ "epoch": 1.1506276150627615,
92
+ "grad_norm": 3.8270370960235596,
93
+ "learning_rate": 1.8493723849372386e-05,
94
+ "loss": 0.1854,
95
+ "step": 275
96
+ },
97
+ {
98
+ "epoch": 1.2552301255230125,
99
+ "grad_norm": 5.27299165725708,
100
+ "learning_rate": 1.7447698744769877e-05,
101
+ "loss": 0.2133,
102
+ "step": 300
103
+ },
104
+ {
105
+ "epoch": 1.3598326359832635,
106
+ "grad_norm": 4.560113906860352,
107
+ "learning_rate": 1.6401673640167365e-05,
108
+ "loss": 0.1894,
109
+ "step": 325
110
+ },
111
+ {
112
+ "epoch": 1.4644351464435146,
113
+ "grad_norm": 2.413438320159912,
114
+ "learning_rate": 1.5355648535564852e-05,
115
+ "loss": 0.1656,
116
+ "step": 350
117
+ },
118
+ {
119
+ "epoch": 1.5690376569037658,
120
+ "grad_norm": 3.1509647369384766,
121
+ "learning_rate": 1.4309623430962344e-05,
122
+ "loss": 0.1924,
123
+ "step": 375
124
+ },
125
+ {
126
+ "epoch": 1.6736401673640167,
127
+ "grad_norm": 4.149273872375488,
128
+ "learning_rate": 1.3263598326359832e-05,
129
+ "loss": 0.1712,
130
+ "step": 400
131
+ },
132
+ {
133
+ "epoch": 1.778242677824268,
134
+ "grad_norm": 2.754230499267578,
135
+ "learning_rate": 1.2217573221757323e-05,
136
+ "loss": 0.2081,
137
+ "step": 425
138
+ },
139
+ {
140
+ "epoch": 1.8828451882845187,
141
+ "grad_norm": 5.342536449432373,
142
+ "learning_rate": 1.1171548117154812e-05,
143
+ "loss": 0.1723,
144
+ "step": 450
145
+ },
146
+ {
147
+ "epoch": 1.98744769874477,
148
+ "grad_norm": 4.396164894104004,
149
+ "learning_rate": 1.0125523012552301e-05,
150
+ "loss": 0.1659,
151
+ "step": 475
152
+ },
153
+ {
154
+ "epoch": 2.0,
155
+ "eval_accuracy": 0.9336702463676564,
156
+ "eval_loss": 0.2127257138490677,
157
+ "eval_runtime": 6.0575,
158
+ "eval_samples_per_second": 261.33,
159
+ "eval_steps_per_second": 2.806,
160
+ "step": 478
161
+ },
162
+ {
163
+ "epoch": 2.092050209205021,
164
+ "grad_norm": 7.640400409698486,
165
+ "learning_rate": 9.079497907949792e-06,
166
+ "loss": 0.1083,
167
+ "step": 500
168
+ },
169
+ {
170
+ "epoch": 2.196652719665272,
171
+ "grad_norm": 7.712110996246338,
172
+ "learning_rate": 8.03347280334728e-06,
173
+ "loss": 0.1285,
174
+ "step": 525
175
+ },
176
+ {
177
+ "epoch": 2.301255230125523,
178
+ "grad_norm": 0.2726390063762665,
179
+ "learning_rate": 6.98744769874477e-06,
180
+ "loss": 0.1304,
181
+ "step": 550
182
+ },
183
+ {
184
+ "epoch": 2.405857740585774,
185
+ "grad_norm": 3.730720043182373,
186
+ "learning_rate": 5.941422594142259e-06,
187
+ "loss": 0.1384,
188
+ "step": 575
189
+ },
190
+ {
191
+ "epoch": 2.510460251046025,
192
+ "grad_norm": 3.225276470184326,
193
+ "learning_rate": 4.895397489539749e-06,
194
+ "loss": 0.1357,
195
+ "step": 600
196
+ },
197
+ {
198
+ "epoch": 2.6150627615062763,
199
+ "grad_norm": 4.274138450622559,
200
+ "learning_rate": 3.849372384937238e-06,
201
+ "loss": 0.126,
202
+ "step": 625
203
+ },
204
+ {
205
+ "epoch": 2.719665271966527,
206
+ "grad_norm": 0.4543670117855072,
207
+ "learning_rate": 2.803347280334728e-06,
208
+ "loss": 0.1263,
209
+ "step": 650
210
+ },
211
+ {
212
+ "epoch": 2.8242677824267783,
213
+ "grad_norm": 3.624878406524658,
214
+ "learning_rate": 1.7573221757322176e-06,
215
+ "loss": 0.137,
216
+ "step": 675
217
+ },
218
+ {
219
+ "epoch": 2.928870292887029,
220
+ "grad_norm": 4.657623291015625,
221
+ "learning_rate": 7.112970711297071e-07,
222
+ "loss": 0.1329,
223
+ "step": 700
224
+ },
225
+ {
226
+ "epoch": 3.0,
227
+ "eval_accuracy": 0.9330385344283006,
228
+ "eval_loss": 0.22735266387462616,
229
+ "eval_runtime": 5.5479,
230
+ "eval_samples_per_second": 285.334,
231
+ "eval_steps_per_second": 3.064,
232
+ "step": 717
233
+ },
234
+ {
235
+ "epoch": 3.0,
236
+ "step": 717,
237
+ "total_flos": 1055495423686536.0,
238
+ "train_loss": 0.20476110576752007,
239
+ "train_runtime": 98.8837,
240
+ "train_samples_per_second": 346.65,
241
+ "train_steps_per_second": 7.251
242
+ }
243
+ ],
244
+ "logging_steps": 25,
245
+ "max_steps": 717,
246
+ "num_input_tokens_seen": 0,
247
+ "num_train_epochs": 3,
248
+ "save_steps": 500,
249
+ "stateful_callbacks": {
250
+ "TrainerControl": {
251
+ "args": {
252
+ "should_epoch_stop": false,
253
+ "should_evaluate": false,
254
+ "should_log": false,
255
+ "should_save": false,
256
+ "should_training_stop": false
257
+ },
258
+ "attributes": {}
259
+ }
260
+ },
261
+ "total_flos": 1055495423686536.0,
262
+ "train_batch_size": 48,
263
+ "trial_name": null,
264
+ "trial_params": null
265
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_44/run_command.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ /usr/local/bin/python /workspace/run_glue_MHA_gated.py --model_name_or_path bert-base-multilingual-cased --dataset_name tmnam20/ViGLUE --task_name vsfc --max_seq_length 128 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 25 --report_to none --num_train_epochs 3 --learning_rate 3e-05 --per_device_train_batch_size 48 --per_device_eval_batch_size 96 --gradient_accumulation_steps 1 --dataloader_num_workers 1 --seed 44 --output_dir result_ablation_mbert/vsfc/mBERT/multi_branch_average/seed_44 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy multi_branch_average --n_heads 4 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05