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

Upload batch 23/24: vsfc/mBERT/mean/seed_43 ... vsfc/mBERT/mha_attention/seed_43

Browse files
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mean/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/mean/seed_43 --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
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mean/seed_44/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.936197094125079,
4
+ "eval_loss": 0.22665680944919586,
5
+ "eval_runtime": 6.7728,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 233.73,
8
+ "eval_steps_per_second": 2.51,
9
+ "total_flos": 1024705701150720.0,
10
+ "train_loss": 0.20191558268612234,
11
+ "train_runtime": 106.7947,
12
+ "train_samples": 11426,
13
+ "train_samples_per_second": 320.971,
14
+ "train_steps_per_second": 6.714
15
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mean/seed_44/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.936197094125079,
4
+ "eval_loss": 0.22665680944919586,
5
+ "eval_runtime": 6.7728,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 233.73,
8
+ "eval_steps_per_second": 2.51
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mean/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/mean/seed_44 --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
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mean/seed_44/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 1024705701150720.0,
4
+ "train_loss": 0.20191558268612234,
5
+ "train_runtime": 106.7947,
6
+ "train_samples": 11426,
7
+ "train_samples_per_second": 320.971,
8
+ "train_steps_per_second": 6.714
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mean/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": 5.314708709716797,
14
+ "learning_rate": 2.8953974895397488e-05,
15
+ "loss": 0.5169,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.20920502092050208,
20
+ "grad_norm": 8.301779747009277,
21
+ "learning_rate": 2.7907949790794982e-05,
22
+ "loss": 0.3416,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.3138075313807531,
27
+ "grad_norm": 5.364657402038574,
28
+ "learning_rate": 2.686192468619247e-05,
29
+ "loss": 0.3466,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 0.41841004184100417,
34
+ "grad_norm": 3.2444639205932617,
35
+ "learning_rate": 2.5815899581589957e-05,
36
+ "loss": 0.2759,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 0.5230125523012552,
41
+ "grad_norm": 6.162262439727783,
42
+ "learning_rate": 2.4769874476987448e-05,
43
+ "loss": 0.2799,
44
+ "step": 125
45
+ },
46
+ {
47
+ "epoch": 0.6276150627615062,
48
+ "grad_norm": 4.790009498596191,
49
+ "learning_rate": 2.372384937238494e-05,
50
+ "loss": 0.2475,
51
+ "step": 150
52
+ },
53
+ {
54
+ "epoch": 0.7322175732217573,
55
+ "grad_norm": 7.008646488189697,
56
+ "learning_rate": 2.2677824267782426e-05,
57
+ "loss": 0.2744,
58
+ "step": 175
59
+ },
60
+ {
61
+ "epoch": 0.8368200836820083,
62
+ "grad_norm": 3.773528575897217,
63
+ "learning_rate": 2.1631799163179917e-05,
64
+ "loss": 0.225,
65
+ "step": 200
66
+ },
67
+ {
68
+ "epoch": 0.9414225941422594,
69
+ "grad_norm": 6.053400993347168,
70
+ "learning_rate": 2.0585774058577408e-05,
71
+ "loss": 0.2334,
72
+ "step": 225
73
+ },
74
+ {
75
+ "epoch": 1.0,
76
+ "eval_accuracy": 0.9286165508528111,
77
+ "eval_loss": 0.22938770055770874,
78
+ "eval_runtime": 6.8869,
79
+ "eval_samples_per_second": 229.855,
80
+ "eval_steps_per_second": 2.468,
81
+ "step": 239
82
+ },
83
+ {
84
+ "epoch": 1.0460251046025104,
85
+ "grad_norm": 5.818469047546387,
86
+ "learning_rate": 1.9539748953974895e-05,
87
+ "loss": 0.203,
88
+ "step": 250
89
+ },
90
+ {
91
+ "epoch": 1.1506276150627615,
92
+ "grad_norm": 8.090319633483887,
93
+ "learning_rate": 1.8493723849372386e-05,
94
+ "loss": 0.1994,
95
+ "step": 275
96
+ },
97
+ {
98
+ "epoch": 1.2552301255230125,
99
+ "grad_norm": 3.0878026485443115,
100
+ "learning_rate": 1.7447698744769877e-05,
101
+ "loss": 0.1816,
102
+ "step": 300
103
+ },
104
+ {
105
+ "epoch": 1.3598326359832635,
106
+ "grad_norm": 3.3793857097625732,
107
+ "learning_rate": 1.6401673640167365e-05,
108
+ "loss": 0.1842,
109
+ "step": 325
110
+ },
111
+ {
112
+ "epoch": 1.4644351464435146,
113
+ "grad_norm": 3.8809702396392822,
114
+ "learning_rate": 1.5355648535564852e-05,
115
+ "loss": 0.1899,
116
+ "step": 350
117
+ },
118
+ {
119
+ "epoch": 1.5690376569037658,
120
+ "grad_norm": 6.736715316772461,
121
+ "learning_rate": 1.4309623430962344e-05,
122
+ "loss": 0.1819,
123
+ "step": 375
124
+ },
125
+ {
126
+ "epoch": 1.6736401673640167,
127
+ "grad_norm": 4.629181861877441,
128
+ "learning_rate": 1.3263598326359832e-05,
129
+ "loss": 0.1905,
130
+ "step": 400
131
+ },
132
+ {
133
+ "epoch": 1.778242677824268,
134
+ "grad_norm": 3.5737199783325195,
135
+ "learning_rate": 1.2217573221757323e-05,
136
+ "loss": 0.1988,
137
+ "step": 425
138
+ },
139
+ {
140
+ "epoch": 1.8828451882845187,
141
+ "grad_norm": 6.036787509918213,
142
+ "learning_rate": 1.1171548117154812e-05,
143
+ "loss": 0.1937,
144
+ "step": 450
145
+ },
146
+ {
147
+ "epoch": 1.98744769874477,
148
+ "grad_norm": 2.8429791927337646,
149
+ "learning_rate": 1.0125523012552301e-05,
150
+ "loss": 0.1477,
151
+ "step": 475
152
+ },
153
+ {
154
+ "epoch": 2.0,
155
+ "eval_accuracy": 0.9317751105495894,
156
+ "eval_loss": 0.22606870532035828,
157
+ "eval_runtime": 6.4536,
158
+ "eval_samples_per_second": 245.291,
159
+ "eval_steps_per_second": 2.634,
160
+ "step": 478
161
+ },
162
+ {
163
+ "epoch": 2.092050209205021,
164
+ "grad_norm": 13.506583213806152,
165
+ "learning_rate": 9.079497907949792e-06,
166
+ "loss": 0.1236,
167
+ "step": 500
168
+ },
169
+ {
170
+ "epoch": 2.196652719665272,
171
+ "grad_norm": 4.173339366912842,
172
+ "learning_rate": 8.03347280334728e-06,
173
+ "loss": 0.1272,
174
+ "step": 525
175
+ },
176
+ {
177
+ "epoch": 2.301255230125523,
178
+ "grad_norm": 2.1523401737213135,
179
+ "learning_rate": 6.98744769874477e-06,
180
+ "loss": 0.1219,
181
+ "step": 550
182
+ },
183
+ {
184
+ "epoch": 2.405857740585774,
185
+ "grad_norm": 4.702078819274902,
186
+ "learning_rate": 5.941422594142259e-06,
187
+ "loss": 0.1243,
188
+ "step": 575
189
+ },
190
+ {
191
+ "epoch": 2.510460251046025,
192
+ "grad_norm": 2.8082313537597656,
193
+ "learning_rate": 4.895397489539749e-06,
194
+ "loss": 0.1243,
195
+ "step": 600
196
+ },
197
+ {
198
+ "epoch": 2.6150627615062763,
199
+ "grad_norm": 6.025881767272949,
200
+ "learning_rate": 3.849372384937238e-06,
201
+ "loss": 0.1141,
202
+ "step": 625
203
+ },
204
+ {
205
+ "epoch": 2.719665271966527,
206
+ "grad_norm": 5.893019199371338,
207
+ "learning_rate": 2.803347280334728e-06,
208
+ "loss": 0.1389,
209
+ "step": 650
210
+ },
211
+ {
212
+ "epoch": 2.8242677824267783,
213
+ "grad_norm": 5.248950481414795,
214
+ "learning_rate": 1.7573221757322176e-06,
215
+ "loss": 0.0995,
216
+ "step": 675
217
+ },
218
+ {
219
+ "epoch": 2.928870292887029,
220
+ "grad_norm": 3.4337127208709717,
221
+ "learning_rate": 7.112970711297071e-07,
222
+ "loss": 0.1231,
223
+ "step": 700
224
+ },
225
+ {
226
+ "epoch": 3.0,
227
+ "eval_accuracy": 0.936197094125079,
228
+ "eval_loss": 0.22665680944919586,
229
+ "eval_runtime": 5.155,
230
+ "eval_samples_per_second": 307.079,
231
+ "eval_steps_per_second": 3.298,
232
+ "step": 717
233
+ },
234
+ {
235
+ "epoch": 3.0,
236
+ "step": 717,
237
+ "total_flos": 1024705701150720.0,
238
+ "train_loss": 0.20191558268612234,
239
+ "train_runtime": 106.7947,
240
+ "train_samples_per_second": 320.971,
241
+ "train_steps_per_second": 6.714
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/mha_attention/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/mha_attention/seed_42 --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/mha_attention/seed_43/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9387239418825016,
4
+ "eval_loss": 0.2242736667394638,
5
+ "eval_runtime": 6.8417,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 231.376,
8
+ "eval_steps_per_second": 2.485,
9
+ "total_flos": 1027433247582168.0,
10
+ "train_loss": 0.20716787132757977,
11
+ "train_runtime": 96.0564,
12
+ "train_samples": 11426,
13
+ "train_samples_per_second": 356.853,
14
+ "train_steps_per_second": 7.464
15
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mha_attention/seed_43/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9387239418825016,
4
+ "eval_loss": 0.2242736667394638,
5
+ "eval_runtime": 6.8417,
6
+ "eval_samples": 1583,
7
+ "eval_samples_per_second": 231.376,
8
+ "eval_steps_per_second": 2.485
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mha_attention/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/mha_attention/seed_43 --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/mha_attention/seed_43/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 1027433247582168.0,
4
+ "train_loss": 0.20716787132757977,
5
+ "train_runtime": 96.0564,
6
+ "train_samples": 11426,
7
+ "train_samples_per_second": 356.853,
8
+ "train_steps_per_second": 7.464
9
+ }
mbert_rtx6000_metrics/result_ablation_mbert/vsfc/mBERT/mha_attention/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.308683395385742,
14
+ "learning_rate": 2.8953974895397488e-05,
15
+ "loss": 0.5152,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.20920502092050208,
20
+ "grad_norm": 7.704500198364258,
21
+ "learning_rate": 2.7907949790794982e-05,
22
+ "loss": 0.3189,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.3138075313807531,
27
+ "grad_norm": 6.910772323608398,
28
+ "learning_rate": 2.686192468619247e-05,
29
+ "loss": 0.3111,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 0.41841004184100417,
34
+ "grad_norm": 4.743614673614502,
35
+ "learning_rate": 2.5815899581589957e-05,
36
+ "loss": 0.2751,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 0.5230125523012552,
41
+ "grad_norm": 10.062735557556152,
42
+ "learning_rate": 2.4769874476987448e-05,
43
+ "loss": 0.3335,
44
+ "step": 125
45
+ },
46
+ {
47
+ "epoch": 0.6276150627615062,
48
+ "grad_norm": 3.8704612255096436,
49
+ "learning_rate": 2.372384937238494e-05,
50
+ "loss": 0.3015,
51
+ "step": 150
52
+ },
53
+ {
54
+ "epoch": 0.7322175732217573,
55
+ "grad_norm": 3.3860716819763184,
56
+ "learning_rate": 2.2677824267782426e-05,
57
+ "loss": 0.2698,
58
+ "step": 175
59
+ },
60
+ {
61
+ "epoch": 0.8368200836820083,
62
+ "grad_norm": 6.553400993347168,
63
+ "learning_rate": 2.1631799163179917e-05,
64
+ "loss": 0.2518,
65
+ "step": 200
66
+ },
67
+ {
68
+ "epoch": 0.9414225941422594,
69
+ "grad_norm": 3.3159918785095215,
70
+ "learning_rate": 2.0585774058577408e-05,
71
+ "loss": 0.2436,
72
+ "step": 225
73
+ },
74
+ {
75
+ "epoch": 1.0,
76
+ "eval_accuracy": 0.9260897030953885,
77
+ "eval_loss": 0.23272140324115753,
78
+ "eval_runtime": 6.213,
79
+ "eval_samples_per_second": 254.79,
80
+ "eval_steps_per_second": 2.736,
81
+ "step": 239
82
+ },
83
+ {
84
+ "epoch": 1.0460251046025104,
85
+ "grad_norm": 5.275418281555176,
86
+ "learning_rate": 1.9539748953974895e-05,
87
+ "loss": 0.2365,
88
+ "step": 250
89
+ },
90
+ {
91
+ "epoch": 1.1506276150627615,
92
+ "grad_norm": 5.208725452423096,
93
+ "learning_rate": 1.8493723849372386e-05,
94
+ "loss": 0.1869,
95
+ "step": 275
96
+ },
97
+ {
98
+ "epoch": 1.2552301255230125,
99
+ "grad_norm": 9.752256393432617,
100
+ "learning_rate": 1.7447698744769877e-05,
101
+ "loss": 0.1835,
102
+ "step": 300
103
+ },
104
+ {
105
+ "epoch": 1.3598326359832635,
106
+ "grad_norm": 11.681489944458008,
107
+ "learning_rate": 1.6401673640167365e-05,
108
+ "loss": 0.1927,
109
+ "step": 325
110
+ },
111
+ {
112
+ "epoch": 1.4644351464435146,
113
+ "grad_norm": 5.413773059844971,
114
+ "learning_rate": 1.5355648535564852e-05,
115
+ "loss": 0.1698,
116
+ "step": 350
117
+ },
118
+ {
119
+ "epoch": 1.5690376569037658,
120
+ "grad_norm": 4.785913944244385,
121
+ "learning_rate": 1.4309623430962344e-05,
122
+ "loss": 0.196,
123
+ "step": 375
124
+ },
125
+ {
126
+ "epoch": 1.6736401673640167,
127
+ "grad_norm": 4.216114044189453,
128
+ "learning_rate": 1.3263598326359832e-05,
129
+ "loss": 0.1911,
130
+ "step": 400
131
+ },
132
+ {
133
+ "epoch": 1.778242677824268,
134
+ "grad_norm": 2.994622230529785,
135
+ "learning_rate": 1.2217573221757323e-05,
136
+ "loss": 0.1976,
137
+ "step": 425
138
+ },
139
+ {
140
+ "epoch": 1.8828451882845187,
141
+ "grad_norm": 4.442307949066162,
142
+ "learning_rate": 1.1171548117154812e-05,
143
+ "loss": 0.1704,
144
+ "step": 450
145
+ },
146
+ {
147
+ "epoch": 1.98744769874477,
148
+ "grad_norm": 6.980963230133057,
149
+ "learning_rate": 1.0125523012552301e-05,
150
+ "loss": 0.1704,
151
+ "step": 475
152
+ },
153
+ {
154
+ "epoch": 2.0,
155
+ "eval_accuracy": 0.934301958307012,
156
+ "eval_loss": 0.21168287098407745,
157
+ "eval_runtime": 6.9147,
158
+ "eval_samples_per_second": 228.934,
159
+ "eval_steps_per_second": 2.459,
160
+ "step": 478
161
+ },
162
+ {
163
+ "epoch": 2.092050209205021,
164
+ "grad_norm": 6.22763204574585,
165
+ "learning_rate": 9.079497907949792e-06,
166
+ "loss": 0.1191,
167
+ "step": 500
168
+ },
169
+ {
170
+ "epoch": 2.196652719665272,
171
+ "grad_norm": 6.816147327423096,
172
+ "learning_rate": 8.03347280334728e-06,
173
+ "loss": 0.1273,
174
+ "step": 525
175
+ },
176
+ {
177
+ "epoch": 2.301255230125523,
178
+ "grad_norm": 0.2829854488372803,
179
+ "learning_rate": 6.98744769874477e-06,
180
+ "loss": 0.1263,
181
+ "step": 550
182
+ },
183
+ {
184
+ "epoch": 2.405857740585774,
185
+ "grad_norm": 6.346888065338135,
186
+ "learning_rate": 5.941422594142259e-06,
187
+ "loss": 0.1425,
188
+ "step": 575
189
+ },
190
+ {
191
+ "epoch": 2.510460251046025,
192
+ "grad_norm": 3.365602731704712,
193
+ "learning_rate": 4.895397489539749e-06,
194
+ "loss": 0.145,
195
+ "step": 600
196
+ },
197
+ {
198
+ "epoch": 2.6150627615062763,
199
+ "grad_norm": 3.679931640625,
200
+ "learning_rate": 3.849372384937238e-06,
201
+ "loss": 0.109,
202
+ "step": 625
203
+ },
204
+ {
205
+ "epoch": 2.719665271966527,
206
+ "grad_norm": 0.2995340824127197,
207
+ "learning_rate": 2.803347280334728e-06,
208
+ "loss": 0.1256,
209
+ "step": 650
210
+ },
211
+ {
212
+ "epoch": 2.8242677824267783,
213
+ "grad_norm": 4.675779342651367,
214
+ "learning_rate": 1.7573221757322176e-06,
215
+ "loss": 0.1201,
216
+ "step": 675
217
+ },
218
+ {
219
+ "epoch": 2.928870292887029,
220
+ "grad_norm": 3.4704439640045166,
221
+ "learning_rate": 7.112970711297071e-07,
222
+ "loss": 0.1452,
223
+ "step": 700
224
+ },
225
+ {
226
+ "epoch": 3.0,
227
+ "eval_accuracy": 0.9387239418825016,
228
+ "eval_loss": 0.2242736667394638,
229
+ "eval_runtime": 5.3531,
230
+ "eval_samples_per_second": 295.716,
231
+ "eval_steps_per_second": 3.176,
232
+ "step": 717
233
+ },
234
+ {
235
+ "epoch": 3.0,
236
+ "step": 717,
237
+ "total_flos": 1027433247582168.0,
238
+ "train_loss": 0.20716787132757977,
239
+ "train_runtime": 96.0564,
240
+ "train_samples_per_second": 356.853,
241
+ "train_steps_per_second": 7.464
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": 1027433247582168.0,
262
+ "train_batch_size": 48,
263
+ "trial_name": null,
264
+ "trial_params": null
265
+ }