PuxAI commited on
Commit
563518c
·
verified ·
1 Parent(s): fabd724

Upload batch 1/14: cola/mBERT/attention/seed_42 ... cola/mBERT/gated_multi_branch/seed_42

Browse files
Files changed (35) hide show
  1. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/all_results.json +15 -0
  2. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/eval_results.json +9 -0
  3. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/run_command.txt +1 -0
  4. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/train_results.json +9 -0
  5. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/trainer_state.json +111 -0
  6. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/all_results.json +15 -0
  7. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/eval_results.json +9 -0
  8. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/run_command.txt +1 -0
  9. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/train_results.json +9 -0
  10. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/trainer_state.json +111 -0
  11. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/all_results.json +15 -0
  12. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/eval_results.json +9 -0
  13. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/run_command.txt +1 -0
  14. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/train_results.json +9 -0
  15. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/trainer_state.json +111 -0
  16. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/all_results.json +15 -0
  17. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/eval_results.json +9 -0
  18. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/run_command.txt +1 -0
  19. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/train_results.json +9 -0
  20. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/trainer_state.json +111 -0
  21. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/all_results.json +15 -0
  22. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/eval_results.json +9 -0
  23. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/run_command.txt +1 -0
  24. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/train_results.json +9 -0
  25. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/trainer_state.json +111 -0
  26. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/all_results.json +15 -0
  27. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/eval_results.json +9 -0
  28. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/run_command.txt +1 -0
  29. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/train_results.json +9 -0
  30. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/trainer_state.json +111 -0
  31. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/all_results.json +15 -0
  32. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/eval_results.json +9 -0
  33. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/run_command.txt +1 -0
  34. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/train_results.json +9 -0
  35. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/trainer_state.json +111 -0
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.3018324375152588,
4
+ "eval_matthews_correlation": 0.7469782739691797,
5
+ "eval_runtime": 1.6368,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 637.222,
8
+ "eval_steps_per_second": 4.277,
9
+ "total_flos": 429108409807626.0,
10
+ "train_loss": 0.24882495366152946,
11
+ "train_runtime": 26.6404,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 962.936,
14
+ "train_steps_per_second": 12.049
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.3018324375152588,
4
+ "eval_matthews_correlation": 0.7469782739691797,
5
+ "eval_runtime": 1.6368,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 637.222,
8
+ "eval_steps_per_second": 4.277
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/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 cola --max_seq_length 256 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 50 --report_to none --num_train_epochs 3 --learning_rate 2e-05 --per_device_train_batch_size 80 --per_device_eval_batch_size 168 --gradient_accumulation_steps 1 --dataloader_num_workers 2 --seed 42 --output_dir result_ablation_mbert_paper/cola/mBERT/attention/seed_42 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy attention --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 429108409807626.0,
4
+ "train_loss": 0.24882495366152946,
5
+ "train_runtime": 26.6404,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 962.936,
8
+ "train_steps_per_second": 12.049
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_42/trainer_state.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 321,
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.4672897196261682,
13
+ "grad_norm": 5.81113862991333,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4141,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 3.5820555686950684,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3117,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.28091415762901306,
28
+ "eval_matthews_correlation": 0.7231377957738392,
29
+ "eval_runtime": 1.6446,
30
+ "eval_samples_per_second": 634.202,
31
+ "eval_steps_per_second": 4.256,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 5.024538516998291,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.25,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 4.812474727630615,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2205,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2870105504989624,
51
+ "eval_matthews_correlation": 0.7566293109993708,
52
+ "eval_runtime": 1.6696,
53
+ "eval_samples_per_second": 624.699,
54
+ "eval_steps_per_second": 4.193,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 6.258780002593994,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.185,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 5.915642738342285,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1552,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.3018324375152588,
74
+ "eval_matthews_correlation": 0.7469782739691797,
75
+ "eval_runtime": 1.4542,
76
+ "eval_samples_per_second": 717.243,
77
+ "eval_steps_per_second": 4.814,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 429108409807626.0,
84
+ "train_loss": 0.24882495366152946,
85
+ "train_runtime": 26.6404,
86
+ "train_samples_per_second": 962.936,
87
+ "train_steps_per_second": 12.049
88
+ }
89
+ ],
90
+ "logging_steps": 50,
91
+ "max_steps": 321,
92
+ "num_input_tokens_seen": 0,
93
+ "num_train_epochs": 3,
94
+ "save_steps": 500,
95
+ "stateful_callbacks": {
96
+ "TrainerControl": {
97
+ "args": {
98
+ "should_epoch_stop": false,
99
+ "should_evaluate": false,
100
+ "should_log": false,
101
+ "should_save": false,
102
+ "should_training_stop": false
103
+ },
104
+ "attributes": {}
105
+ }
106
+ },
107
+ "total_flos": 429108409807626.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.28296852111816406,
4
+ "eval_matthews_correlation": 0.7658812546257014,
5
+ "eval_runtime": 1.422,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 733.484,
8
+ "eval_steps_per_second": 4.923,
9
+ "total_flos": 426225629122602.0,
10
+ "train_loss": 0.26194217346167636,
11
+ "train_runtime": 30.3278,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 845.857,
14
+ "train_steps_per_second": 10.584
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.28296852111816406,
4
+ "eval_matthews_correlation": 0.7658812546257014,
5
+ "eval_runtime": 1.422,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 733.484,
8
+ "eval_steps_per_second": 4.923
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/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 cola --max_seq_length 256 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 50 --report_to none --num_train_epochs 3 --learning_rate 2e-05 --per_device_train_batch_size 80 --per_device_eval_batch_size 168 --gradient_accumulation_steps 1 --dataloader_num_workers 2 --seed 43 --output_dir result_ablation_mbert_paper/cola/mBERT/attention/seed_43 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy attention --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 426225629122602.0,
4
+ "train_loss": 0.26194217346167636,
5
+ "train_runtime": 30.3278,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 845.857,
8
+ "train_steps_per_second": 10.584
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_43/trainer_state.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 321,
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.4672897196261682,
13
+ "grad_norm": 3.5030007362365723,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4648,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 4.771762371063232,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3283,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.2850252389907837,
28
+ "eval_matthews_correlation": 0.7160180415692806,
29
+ "eval_runtime": 1.6694,
30
+ "eval_samples_per_second": 624.763,
31
+ "eval_steps_per_second": 4.193,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 5.225672245025635,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2535,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 4.631227970123291,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2213,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2561209797859192,
51
+ "eval_matthews_correlation": 0.7698455874888194,
52
+ "eval_runtime": 1.5514,
53
+ "eval_samples_per_second": 672.295,
54
+ "eval_steps_per_second": 4.512,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 7.529651641845703,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1778,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 3.7077887058258057,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1662,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.28296852111816406,
74
+ "eval_matthews_correlation": 0.7658812546257014,
75
+ "eval_runtime": 1.2873,
76
+ "eval_samples_per_second": 810.246,
77
+ "eval_steps_per_second": 5.438,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 426225629122602.0,
84
+ "train_loss": 0.26194217346167636,
85
+ "train_runtime": 30.3278,
86
+ "train_samples_per_second": 845.857,
87
+ "train_steps_per_second": 10.584
88
+ }
89
+ ],
90
+ "logging_steps": 50,
91
+ "max_steps": 321,
92
+ "num_input_tokens_seen": 0,
93
+ "num_train_epochs": 3,
94
+ "save_steps": 500,
95
+ "stateful_callbacks": {
96
+ "TrainerControl": {
97
+ "args": {
98
+ "should_epoch_stop": false,
99
+ "should_evaluate": false,
100
+ "should_log": false,
101
+ "should_save": false,
102
+ "should_training_stop": false
103
+ },
104
+ "attributes": {}
105
+ }
106
+ },
107
+ "total_flos": 426225629122602.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.28675714135169983,
4
+ "eval_matthews_correlation": 0.7776576303274508,
5
+ "eval_runtime": 1.3638,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 764.765,
8
+ "eval_steps_per_second": 5.133,
9
+ "total_flos": 426289128768780.0,
10
+ "train_loss": 0.25152042053198886,
11
+ "train_runtime": 30.2705,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 847.46,
14
+ "train_steps_per_second": 10.604
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.28675714135169983,
4
+ "eval_matthews_correlation": 0.7776576303274508,
5
+ "eval_runtime": 1.3638,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 764.765,
8
+ "eval_steps_per_second": 5.133
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/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 cola --max_seq_length 256 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 50 --report_to none --num_train_epochs 3 --learning_rate 2e-05 --per_device_train_batch_size 80 --per_device_eval_batch_size 168 --gradient_accumulation_steps 1 --dataloader_num_workers 2 --seed 44 --output_dir result_ablation_mbert_paper/cola/mBERT/attention/seed_44 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy attention --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 426289128768780.0,
4
+ "train_loss": 0.25152042053198886,
5
+ "train_runtime": 30.2705,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 847.46,
8
+ "train_steps_per_second": 10.604
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/attention/seed_44/trainer_state.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 321,
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.4672897196261682,
13
+ "grad_norm": 4.501162528991699,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4328,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 5.196439266204834,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3116,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.25933611392974854,
28
+ "eval_matthews_correlation": 0.7493314070397258,
29
+ "eval_runtime": 1.6419,
30
+ "eval_samples_per_second": 635.241,
31
+ "eval_steps_per_second": 4.263,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 5.636826992034912,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2497,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 6.54767370223999,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.218,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2662619650363922,
51
+ "eval_matthews_correlation": 0.7823610499496286,
52
+ "eval_runtime": 1.4422,
53
+ "eval_samples_per_second": 723.19,
54
+ "eval_steps_per_second": 4.854,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 9.159810066223145,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1753,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 10.710851669311523,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1671,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.28675714135169983,
74
+ "eval_matthews_correlation": 0.7776576303274508,
75
+ "eval_runtime": 1.3057,
76
+ "eval_samples_per_second": 798.8,
77
+ "eval_steps_per_second": 5.361,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 426289128768780.0,
84
+ "train_loss": 0.25152042053198886,
85
+ "train_runtime": 30.2705,
86
+ "train_samples_per_second": 847.46,
87
+ "train_steps_per_second": 10.604
88
+ }
89
+ ],
90
+ "logging_steps": 50,
91
+ "max_steps": 321,
92
+ "num_input_tokens_seen": 0,
93
+ "num_train_epochs": 3,
94
+ "save_steps": 500,
95
+ "stateful_callbacks": {
96
+ "TrainerControl": {
97
+ "args": {
98
+ "should_epoch_stop": false,
99
+ "should_evaluate": false,
100
+ "should_log": false,
101
+ "should_save": false,
102
+ "should_training_stop": false
103
+ },
104
+ "attributes": {}
105
+ }
106
+ },
107
+ "total_flos": 426289128768780.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.3118951916694641,
4
+ "eval_matthews_correlation": 0.7376962218509274,
5
+ "eval_runtime": 1.3332,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 782.309,
8
+ "eval_steps_per_second": 5.25,
9
+ "total_flos": 429104574692352.0,
10
+ "train_loss": 0.26396365477659994,
11
+ "train_runtime": 34.4335,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 745.0,
14
+ "train_steps_per_second": 9.322
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.3118951916694641,
4
+ "eval_matthews_correlation": 0.7376962218509274,
5
+ "eval_runtime": 1.3332,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 782.309,
8
+ "eval_steps_per_second": 5.25
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/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 cola --max_seq_length 256 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 50 --report_to none --num_train_epochs 3 --learning_rate 2e-05 --per_device_train_batch_size 80 --per_device_eval_batch_size 168 --gradient_accumulation_steps 1 --dataloader_num_workers 2 --seed 42 --output_dir result_ablation_mbert_paper/cola/mBERT/cls/seed_42 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy cls --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 429104574692352.0,
4
+ "train_loss": 0.26396365477659994,
5
+ "train_runtime": 34.4335,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 745.0,
8
+ "train_steps_per_second": 9.322
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_42/trainer_state.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 321,
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.4672897196261682,
13
+ "grad_norm": 5.808596611022949,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4422,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 4.709081649780273,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3206,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.31404176354408264,
28
+ "eval_matthews_correlation": 0.7097825331203318,
29
+ "eval_runtime": 1.5463,
30
+ "eval_samples_per_second": 674.531,
31
+ "eval_steps_per_second": 4.527,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 4.354961395263672,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2704,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 6.4839277267456055,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2277,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.28277787566185,
51
+ "eval_matthews_correlation": 0.73745767326956,
52
+ "eval_runtime": 1.3714,
53
+ "eval_samples_per_second": 760.525,
54
+ "eval_steps_per_second": 5.104,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 7.083432674407959,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.2001,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 5.132438659667969,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1679,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.3118951916694641,
74
+ "eval_matthews_correlation": 0.7376962218509274,
75
+ "eval_runtime": 1.5364,
76
+ "eval_samples_per_second": 678.855,
77
+ "eval_steps_per_second": 4.556,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 429104574692352.0,
84
+ "train_loss": 0.26396365477659994,
85
+ "train_runtime": 34.4335,
86
+ "train_samples_per_second": 745.0,
87
+ "train_steps_per_second": 9.322
88
+ }
89
+ ],
90
+ "logging_steps": 50,
91
+ "max_steps": 321,
92
+ "num_input_tokens_seen": 0,
93
+ "num_train_epochs": 3,
94
+ "save_steps": 500,
95
+ "stateful_callbacks": {
96
+ "TrainerControl": {
97
+ "args": {
98
+ "should_epoch_stop": false,
99
+ "should_evaluate": false,
100
+ "should_log": false,
101
+ "should_save": false,
102
+ "should_training_stop": false
103
+ },
104
+ "attributes": {}
105
+ }
106
+ },
107
+ "total_flos": 429104574692352.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.2848812937736511,
4
+ "eval_matthews_correlation": 0.7682410888691849,
5
+ "eval_runtime": 1.416,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 736.592,
8
+ "eval_steps_per_second": 4.944,
9
+ "total_flos": 426221819771904.0,
10
+ "train_loss": 0.26999439925790947,
11
+ "train_runtime": 34.7235,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 738.78,
14
+ "train_steps_per_second": 9.244
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.2848812937736511,
4
+ "eval_matthews_correlation": 0.7682410888691849,
5
+ "eval_runtime": 1.416,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 736.592,
8
+ "eval_steps_per_second": 4.944
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/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 cola --max_seq_length 256 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 50 --report_to none --num_train_epochs 3 --learning_rate 2e-05 --per_device_train_batch_size 80 --per_device_eval_batch_size 168 --gradient_accumulation_steps 1 --dataloader_num_workers 2 --seed 43 --output_dir result_ablation_mbert_paper/cola/mBERT/cls/seed_43 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy cls --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 426221819771904.0,
4
+ "train_loss": 0.26999439925790947,
5
+ "train_runtime": 34.7235,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 738.78,
8
+ "train_steps_per_second": 9.244
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_43/trainer_state.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 321,
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.4672897196261682,
13
+ "grad_norm": 7.262279987335205,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4513,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 6.623566150665283,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3395,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.2865925431251526,
28
+ "eval_matthews_correlation": 0.7255313529135801,
29
+ "eval_runtime": 1.5661,
30
+ "eval_samples_per_second": 665.996,
31
+ "eval_steps_per_second": 4.47,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 6.386998653411865,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2696,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 5.353734970092773,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2277,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.25154250860214233,
51
+ "eval_matthews_correlation": 0.7925689872887521,
52
+ "eval_runtime": 1.5018,
53
+ "eval_samples_per_second": 694.487,
54
+ "eval_steps_per_second": 4.661,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 6.868773460388184,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1907,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 3.198869466781616,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1804,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.2848812937736511,
74
+ "eval_matthews_correlation": 0.7682410888691849,
75
+ "eval_runtime": 1.4122,
76
+ "eval_samples_per_second": 738.542,
77
+ "eval_steps_per_second": 4.957,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 426221819771904.0,
84
+ "train_loss": 0.26999439925790947,
85
+ "train_runtime": 34.7235,
86
+ "train_samples_per_second": 738.78,
87
+ "train_steps_per_second": 9.244
88
+ }
89
+ ],
90
+ "logging_steps": 50,
91
+ "max_steps": 321,
92
+ "num_input_tokens_seen": 0,
93
+ "num_train_epochs": 3,
94
+ "save_steps": 500,
95
+ "stateful_callbacks": {
96
+ "TrainerControl": {
97
+ "args": {
98
+ "should_epoch_stop": false,
99
+ "should_evaluate": false,
100
+ "should_log": false,
101
+ "should_save": false,
102
+ "should_training_stop": false
103
+ },
104
+ "attributes": {}
105
+ }
106
+ },
107
+ "total_flos": 426221819771904.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.29706358909606934,
4
+ "eval_matthews_correlation": 0.7682418720619295,
5
+ "eval_runtime": 1.3729,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 759.699,
8
+ "eval_steps_per_second": 5.099,
9
+ "total_flos": 426285318850560.0,
10
+ "train_loss": 0.25450210036518417,
11
+ "train_runtime": 34.6579,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 740.177,
14
+ "train_steps_per_second": 9.262
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.29706358909606934,
4
+ "eval_matthews_correlation": 0.7682418720619295,
5
+ "eval_runtime": 1.3729,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 759.699,
8
+ "eval_steps_per_second": 5.099
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/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 cola --max_seq_length 256 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 50 --report_to none --num_train_epochs 3 --learning_rate 2e-05 --per_device_train_batch_size 80 --per_device_eval_batch_size 168 --gradient_accumulation_steps 1 --dataloader_num_workers 2 --seed 44 --output_dir result_ablation_mbert_paper/cola/mBERT/cls/seed_44 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy cls --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 426285318850560.0,
4
+ "train_loss": 0.25450210036518417,
5
+ "train_runtime": 34.6579,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 740.177,
8
+ "train_steps_per_second": 9.262
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/cls/seed_44/trainer_state.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 321,
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.4672897196261682,
13
+ "grad_norm": 5.091413497924805,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4267,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 5.1950764656066895,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.315,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.259922593832016,
28
+ "eval_matthews_correlation": 0.7659095653684762,
29
+ "eval_runtime": 1.5582,
30
+ "eval_samples_per_second": 669.382,
31
+ "eval_steps_per_second": 4.492,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 5.492203712463379,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2544,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 7.6926164627075195,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2234,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2667835056781769,
51
+ "eval_matthews_correlation": 0.7682590021959839,
52
+ "eval_runtime": 1.4626,
53
+ "eval_samples_per_second": 713.094,
54
+ "eval_steps_per_second": 4.786,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 9.207656860351562,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1838,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 11.206198692321777,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1684,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.29706358909606934,
74
+ "eval_matthews_correlation": 0.7682418720619295,
75
+ "eval_runtime": 1.3802,
76
+ "eval_samples_per_second": 755.693,
77
+ "eval_steps_per_second": 5.072,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 426285318850560.0,
84
+ "train_loss": 0.25450210036518417,
85
+ "train_runtime": 34.6579,
86
+ "train_samples_per_second": 740.177,
87
+ "train_steps_per_second": 9.262
88
+ }
89
+ ],
90
+ "logging_steps": 50,
91
+ "max_steps": 321,
92
+ "num_input_tokens_seen": 0,
93
+ "num_train_epochs": 3,
94
+ "save_steps": 500,
95
+ "stateful_callbacks": {
96
+ "TrainerControl": {
97
+ "args": {
98
+ "should_epoch_stop": false,
99
+ "should_evaluate": false,
100
+ "should_log": false,
101
+ "should_save": false,
102
+ "should_training_stop": false
103
+ },
104
+ "attributes": {}
105
+ }
106
+ },
107
+ "total_flos": 426285318850560.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.30131927132606506,
4
+ "eval_matthews_correlation": 0.7641385489211314,
5
+ "eval_runtime": 1.5389,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 677.745,
8
+ "eval_steps_per_second": 4.549,
9
+ "total_flos": 448737841399356.0,
10
+ "train_loss": 0.2518068837970959,
11
+ "train_runtime": 24.6142,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 1042.203,
14
+ "train_steps_per_second": 13.041
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.30131927132606506,
4
+ "eval_matthews_correlation": 0.7641385489211314,
5
+ "eval_runtime": 1.5389,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 677.745,
8
+ "eval_steps_per_second": 4.549
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/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 cola --max_seq_length 256 --do_train --do_eval --evaluation_strategy epoch --save_strategy no --logging_strategy steps --logging_steps 50 --report_to none --num_train_epochs 3 --learning_rate 2e-05 --per_device_train_batch_size 80 --per_device_eval_batch_size 168 --gradient_accumulation_steps 1 --dataloader_num_workers 2 --seed 42 --output_dir result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy gated_multi_branch --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 448737841399356.0,
4
+ "train_loss": 0.2518068837970959,
5
+ "train_runtime": 24.6142,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 1042.203,
8
+ "train_steps_per_second": 13.041
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_42/trainer_state.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 321,
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.4672897196261682,
13
+ "grad_norm": 4.675717353820801,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4265,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 5.895548343658447,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3141,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.2743222415447235,
28
+ "eval_matthews_correlation": 0.7231769633837785,
29
+ "eval_runtime": 1.4946,
30
+ "eval_samples_per_second": 697.848,
31
+ "eval_steps_per_second": 4.684,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 8.380515098571777,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2507,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 4.941604137420654,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2166,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.25941094756126404,
51
+ "eval_matthews_correlation": 0.7706478059846233,
52
+ "eval_runtime": 1.3362,
53
+ "eval_samples_per_second": 780.59,
54
+ "eval_steps_per_second": 5.239,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 4.413182735443115,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1945,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 7.304749488830566,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1546,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.30131927132606506,
74
+ "eval_matthews_correlation": 0.7641385489211314,
75
+ "eval_runtime": 1.6292,
76
+ "eval_samples_per_second": 640.191,
77
+ "eval_steps_per_second": 4.297,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 448737841399356.0,
84
+ "train_loss": 0.2518068837970959,
85
+ "train_runtime": 24.6142,
86
+ "train_samples_per_second": 1042.203,
87
+ "train_steps_per_second": 13.041
88
+ }
89
+ ],
90
+ "logging_steps": 50,
91
+ "max_steps": 321,
92
+ "num_input_tokens_seen": 0,
93
+ "num_train_epochs": 3,
94
+ "save_steps": 500,
95
+ "stateful_callbacks": {
96
+ "TrainerControl": {
97
+ "args": {
98
+ "should_epoch_stop": false,
99
+ "should_evaluate": false,
100
+ "should_log": false,
101
+ "should_save": false,
102
+ "should_training_stop": false
103
+ },
104
+ "attributes": {}
105
+ }
106
+ },
107
+ "total_flos": 448737841399356.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }