PuxAI commited on
Commit
8d5418e
·
verified ·
1 Parent(s): 5c5789d

Upload batch 22/24: vsfc/mBERT/max/seed_42 ... vsfc/mBERT/mean/seed_42

Browse files
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/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/max/seed_42 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy max --n_heads 4 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/seed_43/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9317751105495894,
4
+ "eval_loss": 0.23970364034175873,
5
+ "eval_runtime": 5.8795,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 269.241,
8
+ "eval_steps_per_second": 2.891,
9
+ "total_flos": 1022750124392448.0,
10
+ "train_loss": 0.20938005839764157,
11
+ "train_runtime": 114.2677,
12
+ "train_samples": 11426,
13
+ "train_samples_per_second": 299.98,
14
+ "train_steps_per_second": 6.275
15
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/seed_43/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9317751105495894,
4
+ "eval_loss": 0.23970364034175873,
5
+ "eval_runtime": 5.8795,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 269.241,
8
+ "eval_steps_per_second": 2.891
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/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/max/seed_43 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy max --n_heads 4 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/seed_43/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 1022750124392448.0,
4
+ "train_loss": 0.20938005839764157,
5
+ "train_runtime": 114.2677,
6
+ "train_samples": 11426,
7
+ "train_samples_per_second": 299.98,
8
+ "train_steps_per_second": 6.275
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/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": 6.728825092315674,
14
+ "learning_rate": 2.8953974895397488e-05,
15
+ "loss": 0.5651,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.20920502092050208,
20
+ "grad_norm": 7.908098220825195,
21
+ "learning_rate": 2.7907949790794982e-05,
22
+ "loss": 0.3427,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.3138075313807531,
27
+ "grad_norm": 5.102977752685547,
28
+ "learning_rate": 2.686192468619247e-05,
29
+ "loss": 0.3305,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 0.41841004184100417,
34
+ "grad_norm": 3.49076247215271,
35
+ "learning_rate": 2.5815899581589957e-05,
36
+ "loss": 0.3034,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 0.5230125523012552,
41
+ "grad_norm": 8.612030029296875,
42
+ "learning_rate": 2.4769874476987448e-05,
43
+ "loss": 0.3254,
44
+ "step": 125
45
+ },
46
+ {
47
+ "epoch": 0.6276150627615062,
48
+ "grad_norm": 3.783565044403076,
49
+ "learning_rate": 2.372384937238494e-05,
50
+ "loss": 0.2945,
51
+ "step": 150
52
+ },
53
+ {
54
+ "epoch": 0.7322175732217573,
55
+ "grad_norm": 4.309210777282715,
56
+ "learning_rate": 2.2677824267782426e-05,
57
+ "loss": 0.264,
58
+ "step": 175
59
+ },
60
+ {
61
+ "epoch": 0.8368200836820083,
62
+ "grad_norm": 4.5513763427734375,
63
+ "learning_rate": 2.1631799163179917e-05,
64
+ "loss": 0.2276,
65
+ "step": 200
66
+ },
67
+ {
68
+ "epoch": 0.9414225941422594,
69
+ "grad_norm": 4.058280944824219,
70
+ "learning_rate": 2.0585774058577408e-05,
71
+ "loss": 0.2444,
72
+ "step": 225
73
+ },
74
+ {
75
+ "epoch": 1.0,
76
+ "eval_accuracy": 0.9248262792166771,
77
+ "eval_loss": 0.22704137861728668,
78
+ "eval_runtime": 6.82,
79
+ "eval_samples_per_second": 232.113,
80
+ "eval_steps_per_second": 2.493,
81
+ "step": 239
82
+ },
83
+ {
84
+ "epoch": 1.0460251046025104,
85
+ "grad_norm": 3.737504720687866,
86
+ "learning_rate": 1.9539748953974895e-05,
87
+ "loss": 0.2356,
88
+ "step": 250
89
+ },
90
+ {
91
+ "epoch": 1.1506276150627615,
92
+ "grad_norm": 3.996934652328491,
93
+ "learning_rate": 1.8493723849372386e-05,
94
+ "loss": 0.1916,
95
+ "step": 275
96
+ },
97
+ {
98
+ "epoch": 1.2552301255230125,
99
+ "grad_norm": 6.380361557006836,
100
+ "learning_rate": 1.7447698744769877e-05,
101
+ "loss": 0.2191,
102
+ "step": 300
103
+ },
104
+ {
105
+ "epoch": 1.3598326359832635,
106
+ "grad_norm": 9.467670440673828,
107
+ "learning_rate": 1.6401673640167365e-05,
108
+ "loss": 0.1908,
109
+ "step": 325
110
+ },
111
+ {
112
+ "epoch": 1.4644351464435146,
113
+ "grad_norm": 3.292236566543579,
114
+ "learning_rate": 1.5355648535564852e-05,
115
+ "loss": 0.1608,
116
+ "step": 350
117
+ },
118
+ {
119
+ "epoch": 1.5690376569037658,
120
+ "grad_norm": 3.0692200660705566,
121
+ "learning_rate": 1.4309623430962344e-05,
122
+ "loss": 0.1928,
123
+ "step": 375
124
+ },
125
+ {
126
+ "epoch": 1.6736401673640167,
127
+ "grad_norm": 4.897969722747803,
128
+ "learning_rate": 1.3263598326359832e-05,
129
+ "loss": 0.1785,
130
+ "step": 400
131
+ },
132
+ {
133
+ "epoch": 1.778242677824268,
134
+ "grad_norm": 13.100573539733887,
135
+ "learning_rate": 1.2217573221757323e-05,
136
+ "loss": 0.1813,
137
+ "step": 425
138
+ },
139
+ {
140
+ "epoch": 1.8828451882845187,
141
+ "grad_norm": 3.577944755554199,
142
+ "learning_rate": 1.1171548117154812e-05,
143
+ "loss": 0.166,
144
+ "step": 450
145
+ },
146
+ {
147
+ "epoch": 1.98744769874477,
148
+ "grad_norm": 3.756112575531006,
149
+ "learning_rate": 1.0125523012552301e-05,
150
+ "loss": 0.177,
151
+ "step": 475
152
+ },
153
+ {
154
+ "epoch": 2.0,
155
+ "eval_accuracy": 0.934301958307012,
156
+ "eval_loss": 0.22262610495090485,
157
+ "eval_runtime": 6.3532,
158
+ "eval_samples_per_second": 249.164,
159
+ "eval_steps_per_second": 2.676,
160
+ "step": 478
161
+ },
162
+ {
163
+ "epoch": 2.092050209205021,
164
+ "grad_norm": 7.469801425933838,
165
+ "learning_rate": 9.079497907949792e-06,
166
+ "loss": 0.1089,
167
+ "step": 500
168
+ },
169
+ {
170
+ "epoch": 2.196652719665272,
171
+ "grad_norm": 8.408180236816406,
172
+ "learning_rate": 8.03347280334728e-06,
173
+ "loss": 0.1333,
174
+ "step": 525
175
+ },
176
+ {
177
+ "epoch": 2.301255230125523,
178
+ "grad_norm": 0.488555908203125,
179
+ "learning_rate": 6.98744769874477e-06,
180
+ "loss": 0.1267,
181
+ "step": 550
182
+ },
183
+ {
184
+ "epoch": 2.405857740585774,
185
+ "grad_norm": 5.445498943328857,
186
+ "learning_rate": 5.941422594142259e-06,
187
+ "loss": 0.1297,
188
+ "step": 575
189
+ },
190
+ {
191
+ "epoch": 2.510460251046025,
192
+ "grad_norm": 4.109771728515625,
193
+ "learning_rate": 4.895397489539749e-06,
194
+ "loss": 0.1332,
195
+ "step": 600
196
+ },
197
+ {
198
+ "epoch": 2.6150627615062763,
199
+ "grad_norm": 3.7967689037323,
200
+ "learning_rate": 3.849372384937238e-06,
201
+ "loss": 0.1096,
202
+ "step": 625
203
+ },
204
+ {
205
+ "epoch": 2.719665271966527,
206
+ "grad_norm": 0.5358346700668335,
207
+ "learning_rate": 2.803347280334728e-06,
208
+ "loss": 0.1369,
209
+ "step": 650
210
+ },
211
+ {
212
+ "epoch": 2.8242677824267783,
213
+ "grad_norm": 4.2060651779174805,
214
+ "learning_rate": 1.7573221757322176e-06,
215
+ "loss": 0.1277,
216
+ "step": 675
217
+ },
218
+ {
219
+ "epoch": 2.928870292887029,
220
+ "grad_norm": 4.618159770965576,
221
+ "learning_rate": 7.112970711297071e-07,
222
+ "loss": 0.1439,
223
+ "step": 700
224
+ },
225
+ {
226
+ "epoch": 3.0,
227
+ "eval_accuracy": 0.9317751105495894,
228
+ "eval_loss": 0.23970364034175873,
229
+ "eval_runtime": 5.5616,
230
+ "eval_samples_per_second": 284.628,
231
+ "eval_steps_per_second": 3.057,
232
+ "step": 717
233
+ },
234
+ {
235
+ "epoch": 3.0,
236
+ "step": 717,
237
+ "total_flos": 1022750124392448.0,
238
+ "train_loss": 0.20938005839764157,
239
+ "train_runtime": 114.2677,
240
+ "train_samples_per_second": 299.98,
241
+ "train_steps_per_second": 6.275
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": 1022750124392448.0,
262
+ "train_batch_size": 48,
263
+ "trial_name": null,
264
+ "trial_params": null
265
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/seed_44/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9330385344283006,
4
+ "eval_loss": 0.2280229926109314,
5
+ "eval_runtime": 6.1519,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 257.32,
8
+ "eval_steps_per_second": 2.763,
9
+ "total_flos": 1024705701150720.0,
10
+ "train_loss": 0.20747514532533484,
11
+ "train_runtime": 114.3541,
12
+ "train_samples": 11426,
13
+ "train_samples_per_second": 299.753,
14
+ "train_steps_per_second": 6.27
15
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/seed_44/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9330385344283006,
4
+ "eval_loss": 0.2280229926109314,
5
+ "eval_runtime": 6.1519,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 257.32,
8
+ "eval_steps_per_second": 2.763
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/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/max/seed_44 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy max --n_heads 4 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/seed_44/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 1024705701150720.0,
4
+ "train_loss": 0.20747514532533484,
5
+ "train_runtime": 114.3541,
6
+ "train_samples": 11426,
7
+ "train_samples_per_second": 299.753,
8
+ "train_steps_per_second": 6.27
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/max/seed_44/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": 6.769598484039307,
14
+ "learning_rate": 2.8953974895397488e-05,
15
+ "loss": 0.5558,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.20920502092050208,
20
+ "grad_norm": 4.832651138305664,
21
+ "learning_rate": 2.7907949790794982e-05,
22
+ "loss": 0.3485,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.3138075313807531,
27
+ "grad_norm": 4.488345623016357,
28
+ "learning_rate": 2.686192468619247e-05,
29
+ "loss": 0.3463,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 0.41841004184100417,
34
+ "grad_norm": 2.1114728450775146,
35
+ "learning_rate": 2.5815899581589957e-05,
36
+ "loss": 0.279,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 0.5230125523012552,
41
+ "grad_norm": 5.334822177886963,
42
+ "learning_rate": 2.4769874476987448e-05,
43
+ "loss": 0.2882,
44
+ "step": 125
45
+ },
46
+ {
47
+ "epoch": 0.6276150627615062,
48
+ "grad_norm": 4.173310279846191,
49
+ "learning_rate": 2.372384937238494e-05,
50
+ "loss": 0.2539,
51
+ "step": 150
52
+ },
53
+ {
54
+ "epoch": 0.7322175732217573,
55
+ "grad_norm": 7.422400951385498,
56
+ "learning_rate": 2.2677824267782426e-05,
57
+ "loss": 0.3082,
58
+ "step": 175
59
+ },
60
+ {
61
+ "epoch": 0.8368200836820083,
62
+ "grad_norm": 4.0469560623168945,
63
+ "learning_rate": 2.1631799163179917e-05,
64
+ "loss": 0.22,
65
+ "step": 200
66
+ },
67
+ {
68
+ "epoch": 0.9414225941422594,
69
+ "grad_norm": 6.489193439483643,
70
+ "learning_rate": 2.0585774058577408e-05,
71
+ "loss": 0.2361,
72
+ "step": 225
73
+ },
74
+ {
75
+ "epoch": 1.0,
76
+ "eval_accuracy": 0.9216677195198989,
77
+ "eval_loss": 0.23060229420661926,
78
+ "eval_runtime": 6.7396,
79
+ "eval_samples_per_second": 234.882,
80
+ "eval_steps_per_second": 2.522,
81
+ "step": 239
82
+ },
83
+ {
84
+ "epoch": 1.0460251046025104,
85
+ "grad_norm": 5.142463684082031,
86
+ "learning_rate": 1.9539748953974895e-05,
87
+ "loss": 0.1979,
88
+ "step": 250
89
+ },
90
+ {
91
+ "epoch": 1.1506276150627615,
92
+ "grad_norm": 7.232017993927002,
93
+ "learning_rate": 1.8493723849372386e-05,
94
+ "loss": 0.1866,
95
+ "step": 275
96
+ },
97
+ {
98
+ "epoch": 1.2552301255230125,
99
+ "grad_norm": 3.043032169342041,
100
+ "learning_rate": 1.7447698744769877e-05,
101
+ "loss": 0.1892,
102
+ "step": 300
103
+ },
104
+ {
105
+ "epoch": 1.3598326359832635,
106
+ "grad_norm": 6.970592021942139,
107
+ "learning_rate": 1.6401673640167365e-05,
108
+ "loss": 0.1975,
109
+ "step": 325
110
+ },
111
+ {
112
+ "epoch": 1.4644351464435146,
113
+ "grad_norm": 4.755401611328125,
114
+ "learning_rate": 1.5355648535564852e-05,
115
+ "loss": 0.1931,
116
+ "step": 350
117
+ },
118
+ {
119
+ "epoch": 1.5690376569037658,
120
+ "grad_norm": 7.054227828979492,
121
+ "learning_rate": 1.4309623430962344e-05,
122
+ "loss": 0.179,
123
+ "step": 375
124
+ },
125
+ {
126
+ "epoch": 1.6736401673640167,
127
+ "grad_norm": 6.783203601837158,
128
+ "learning_rate": 1.3263598326359832e-05,
129
+ "loss": 0.1963,
130
+ "step": 400
131
+ },
132
+ {
133
+ "epoch": 1.778242677824268,
134
+ "grad_norm": 3.4103643894195557,
135
+ "learning_rate": 1.2217573221757323e-05,
136
+ "loss": 0.1996,
137
+ "step": 425
138
+ },
139
+ {
140
+ "epoch": 1.8828451882845187,
141
+ "grad_norm": 6.070463180541992,
142
+ "learning_rate": 1.1171548117154812e-05,
143
+ "loss": 0.1977,
144
+ "step": 450
145
+ },
146
+ {
147
+ "epoch": 1.98744769874477,
148
+ "grad_norm": 3.7371246814727783,
149
+ "learning_rate": 1.0125523012552301e-05,
150
+ "loss": 0.1638,
151
+ "step": 475
152
+ },
153
+ {
154
+ "epoch": 2.0,
155
+ "eval_accuracy": 0.9330385344283006,
156
+ "eval_loss": 0.23419837653636932,
157
+ "eval_runtime": 6.5683,
158
+ "eval_samples_per_second": 241.007,
159
+ "eval_steps_per_second": 2.588,
160
+ "step": 478
161
+ },
162
+ {
163
+ "epoch": 2.092050209205021,
164
+ "grad_norm": 9.284645080566406,
165
+ "learning_rate": 9.079497907949792e-06,
166
+ "loss": 0.1358,
167
+ "step": 500
168
+ },
169
+ {
170
+ "epoch": 2.196652719665272,
171
+ "grad_norm": 3.2133939266204834,
172
+ "learning_rate": 8.03347280334728e-06,
173
+ "loss": 0.1323,
174
+ "step": 525
175
+ },
176
+ {
177
+ "epoch": 2.301255230125523,
178
+ "grad_norm": 2.1595170497894287,
179
+ "learning_rate": 6.98744769874477e-06,
180
+ "loss": 0.1245,
181
+ "step": 550
182
+ },
183
+ {
184
+ "epoch": 2.405857740585774,
185
+ "grad_norm": 3.053664207458496,
186
+ "learning_rate": 5.941422594142259e-06,
187
+ "loss": 0.1292,
188
+ "step": 575
189
+ },
190
+ {
191
+ "epoch": 2.510460251046025,
192
+ "grad_norm": 2.4191031455993652,
193
+ "learning_rate": 4.895397489539749e-06,
194
+ "loss": 0.11,
195
+ "step": 600
196
+ },
197
+ {
198
+ "epoch": 2.6150627615062763,
199
+ "grad_norm": 4.869324207305908,
200
+ "learning_rate": 3.849372384937238e-06,
201
+ "loss": 0.1259,
202
+ "step": 625
203
+ },
204
+ {
205
+ "epoch": 2.719665271966527,
206
+ "grad_norm": 4.810281276702881,
207
+ "learning_rate": 2.803347280334728e-06,
208
+ "loss": 0.1491,
209
+ "step": 650
210
+ },
211
+ {
212
+ "epoch": 2.8242677824267783,
213
+ "grad_norm": 3.5581533908843994,
214
+ "learning_rate": 1.7573221757322176e-06,
215
+ "loss": 0.1066,
216
+ "step": 675
217
+ },
218
+ {
219
+ "epoch": 2.928870292887029,
220
+ "grad_norm": 3.9768917560577393,
221
+ "learning_rate": 7.112970711297071e-07,
222
+ "loss": 0.1213,
223
+ "step": 700
224
+ },
225
+ {
226
+ "epoch": 3.0,
227
+ "eval_accuracy": 0.9330385344283006,
228
+ "eval_loss": 0.2280229926109314,
229
+ "eval_runtime": 5.5941,
230
+ "eval_samples_per_second": 282.978,
231
+ "eval_steps_per_second": 3.039,
232
+ "step": 717
233
+ },
234
+ {
235
+ "epoch": 3.0,
236
+ "step": 717,
237
+ "total_flos": 1024705701150720.0,
238
+ "train_loss": 0.20747514532533484,
239
+ "train_runtime": 114.3541,
240
+ "train_samples_per_second": 299.753,
241
+ "train_steps_per_second": 6.27
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": 1024705701150720.0,
262
+ "train_batch_size": 48,
263
+ "trial_name": null,
264
+ "trial_params": null
265
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mean/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/mean/seed_42 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy mean --n_heads 4 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05