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

Upload batch 2/14: cola/mBERT/gated_multi_branch/seed_43 ... cola/mBERT/max/seed_43

Browse files
Files changed (35) hide show
  1. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_43/all_results.json +15 -0
  2. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_43/eval_results.json +9 -0
  3. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_43/run_command.txt +1 -0
  4. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_43/train_results.json +9 -0
  5. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_43/trainer_state.json +111 -0
  6. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_44/all_results.json +15 -0
  7. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_44/eval_results.json +9 -0
  8. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_44/run_command.txt +1 -0
  9. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_44/train_results.json +9 -0
  10. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_44/trainer_state.json +111 -0
  11. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_42/all_results.json +15 -0
  12. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_42/eval_results.json +9 -0
  13. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_42/run_command.txt +1 -0
  14. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_42/train_results.json +9 -0
  15. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_42/trainer_state.json +111 -0
  16. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_43/all_results.json +15 -0
  17. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_43/eval_results.json +9 -0
  18. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_43/run_command.txt +1 -0
  19. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_43/train_results.json +9 -0
  20. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_43/trainer_state.json +111 -0
  21. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_44/all_results.json +15 -0
  22. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_44/eval_results.json +9 -0
  23. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_44/run_command.txt +1 -0
  24. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_44/train_results.json +9 -0
  25. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_44/trainer_state.json +111 -0
  26. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_42/all_results.json +15 -0
  27. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_42/eval_results.json +9 -0
  28. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_42/run_command.txt +1 -0
  29. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_42/train_results.json +9 -0
  30. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_42/trainer_state.json +111 -0
  31. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_43/all_results.json +15 -0
  32. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_43/eval_results.json +9 -0
  33. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_43/run_command.txt +1 -0
  34. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_43/train_results.json +9 -0
  35. mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_43/trainer_state.json +111 -0
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_43/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.29004186391830444,
4
+ "eval_matthews_correlation": 0.7733151048821603,
5
+ "eval_runtime": 1.3984,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 745.833,
8
+ "eval_steps_per_second": 5.006,
9
+ "total_flos": 445723188802812.0,
10
+ "train_loss": 0.2568637455735251,
11
+ "train_runtime": 31.4305,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 816.183,
14
+ "train_steps_per_second": 10.213
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_43/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.29004186391830444,
4
+ "eval_matthews_correlation": 0.7733151048821603,
5
+ "eval_runtime": 1.3984,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 745.833,
8
+ "eval_steps_per_second": 5.006
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/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/gated_multi_branch/seed_43 --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_43/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 445723188802812.0,
4
+ "train_loss": 0.2568637455735251,
5
+ "train_runtime": 31.4305,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 816.183,
8
+ "train_steps_per_second": 10.213
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/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": 5.017457008361816,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4265,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 5.256691932678223,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3242,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.29493391513824463,
28
+ "eval_matthews_correlation": 0.7000821033544287,
29
+ "eval_runtime": 1.4881,
30
+ "eval_samples_per_second": 700.888,
31
+ "eval_steps_per_second": 4.704,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 7.021495819091797,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2655,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 5.001773834228516,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2186,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.238822340965271,
51
+ "eval_matthews_correlation": 0.7808239264042255,
52
+ "eval_runtime": 1.3022,
53
+ "eval_samples_per_second": 800.951,
54
+ "eval_steps_per_second": 5.376,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 5.757174968719482,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1859,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 5.819880485534668,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1647,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.29004186391830444,
74
+ "eval_matthews_correlation": 0.7733151048821603,
75
+ "eval_runtime": 1.5837,
76
+ "eval_samples_per_second": 658.588,
77
+ "eval_steps_per_second": 4.42,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 445723188802812.0,
84
+ "train_loss": 0.2568637455735251,
85
+ "train_runtime": 31.4305,
86
+ "train_samples_per_second": 816.183,
87
+ "train_steps_per_second": 10.213
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": 445723188802812.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_44/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.2971993386745453,
4
+ "eval_matthews_correlation": 0.7612027683763856,
5
+ "eval_runtime": 1.442,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 723.28,
8
+ "eval_steps_per_second": 4.854,
9
+ "total_flos": 445789593220680.0,
10
+ "train_loss": 0.24624314412149684,
11
+ "train_runtime": 31.4852,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 814.765,
14
+ "train_steps_per_second": 10.195
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/seed_44/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.2971993386745453,
4
+ "eval_matthews_correlation": 0.7612027683763856,
5
+ "eval_runtime": 1.442,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 723.28,
8
+ "eval_steps_per_second": 4.854
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/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/gated_multi_branch/seed_44 --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_44/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 445789593220680.0,
4
+ "train_loss": 0.24624314412149684,
5
+ "train_runtime": 31.4852,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 814.765,
8
+ "train_steps_per_second": 10.195
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/gated_multi_branch/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": 6.518551349639893,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4293,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 5.744719982147217,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3058,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.25964823365211487,
28
+ "eval_matthews_correlation": 0.7445807376620748,
29
+ "eval_runtime": 1.5036,
30
+ "eval_samples_per_second": 693.649,
31
+ "eval_steps_per_second": 4.655,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 4.169808864593506,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2458,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 6.322945594787598,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2151,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2919430136680603,
51
+ "eval_matthews_correlation": 0.7310312837735538,
52
+ "eval_runtime": 1.3657,
53
+ "eval_samples_per_second": 763.721,
54
+ "eval_steps_per_second": 5.126,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 5.181381702423096,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1691,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 6.789882183074951,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1584,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.2971993386745453,
74
+ "eval_matthews_correlation": 0.7612027683763856,
75
+ "eval_runtime": 1.4424,
76
+ "eval_samples_per_second": 723.11,
77
+ "eval_steps_per_second": 4.853,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 445789593220680.0,
84
+ "train_loss": 0.24624314412149684,
85
+ "train_runtime": 31.4852,
86
+ "train_samples_per_second": 814.765,
87
+ "train_steps_per_second": 10.195
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": 445789593220680.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_42/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.29159754514694214,
4
+ "eval_matthews_correlation": 0.7493368300485673,
5
+ "eval_runtime": 1.4191,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 734.973,
8
+ "eval_steps_per_second": 4.933,
9
+ "total_flos": 427139728936980.0,
10
+ "train_loss": 0.2649622825075904,
11
+ "train_runtime": 35.4989,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 722.643,
14
+ "train_steps_per_second": 9.043
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_42/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.29159754514694214,
4
+ "eval_matthews_correlation": 0.7493368300485673,
5
+ "eval_runtime": 1.4191,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 734.973,
8
+ "eval_steps_per_second": 4.933
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/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/hf_sequence_classifier/seed_42 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_42/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 427139728936980.0,
4
+ "train_loss": 0.2649622825075904,
5
+ "train_runtime": 35.4989,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 722.643,
8
+ "train_steps_per_second": 9.043
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/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.549562931060791,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4495,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 4.5690693855285645,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3183,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.2829677164554596,
28
+ "eval_matthews_correlation": 0.7168798974635037,
29
+ "eval_runtime": 1.6118,
30
+ "eval_samples_per_second": 647.108,
31
+ "eval_steps_per_second": 4.343,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 4.647832870483398,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.269,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 5.157256603240967,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2311,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2866717576980591,
51
+ "eval_matthews_correlation": 0.7448734653696512,
52
+ "eval_runtime": 1.5508,
53
+ "eval_samples_per_second": 672.555,
54
+ "eval_steps_per_second": 4.514,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 6.212674140930176,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1985,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 5.383189678192139,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1705,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.29159754514694214,
74
+ "eval_matthews_correlation": 0.7493368300485673,
75
+ "eval_runtime": 1.4308,
76
+ "eval_samples_per_second": 728.943,
77
+ "eval_steps_per_second": 4.892,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 427139728936980.0,
84
+ "train_loss": 0.2649622825075904,
85
+ "train_runtime": 35.4989,
86
+ "train_samples_per_second": 722.643,
87
+ "train_steps_per_second": 9.043
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": 427139728936980.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_43/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.29299455881118774,
4
+ "eval_matthews_correlation": 0.770612811219967,
5
+ "eval_runtime": 1.4686,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 710.195,
8
+ "eval_steps_per_second": 4.766,
9
+ "total_flos": 424270173989460.0,
10
+ "train_loss": 0.2762503274876009,
11
+ "train_runtime": 35.5792,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 721.011,
14
+ "train_steps_per_second": 9.022
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_43/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.29299455881118774,
4
+ "eval_matthews_correlation": 0.770612811219967,
5
+ "eval_runtime": 1.4686,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 710.195,
8
+ "eval_steps_per_second": 4.766
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/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/hf_sequence_classifier/seed_43 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_43/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 424270173989460.0,
4
+ "train_loss": 0.2762503274876009,
5
+ "train_runtime": 35.5792,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 721.011,
8
+ "train_steps_per_second": 9.022
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/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": 6.983142375946045,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4794,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 5.011570930480957,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3407,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.2928970158100128,
28
+ "eval_matthews_correlation": 0.7255589981043564,
29
+ "eval_runtime": 1.5391,
30
+ "eval_samples_per_second": 677.655,
31
+ "eval_steps_per_second": 4.548,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 5.626948356628418,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.271,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 4.365847110748291,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.231,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2618351876735687,
51
+ "eval_matthews_correlation": 0.7756438036656587,
52
+ "eval_runtime": 1.5241,
53
+ "eval_samples_per_second": 684.35,
54
+ "eval_steps_per_second": 4.593,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 7.09475564956665,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.196,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 4.770914554595947,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1815,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.29299455881118774,
74
+ "eval_matthews_correlation": 0.770612811219967,
75
+ "eval_runtime": 1.5086,
76
+ "eval_samples_per_second": 691.356,
77
+ "eval_steps_per_second": 4.64,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 424270173989460.0,
84
+ "train_loss": 0.2762503274876009,
85
+ "train_runtime": 35.5792,
86
+ "train_samples_per_second": 721.011,
87
+ "train_steps_per_second": 9.022
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": 424270173989460.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_44/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.2829234004020691,
4
+ "eval_matthews_correlation": 0.7729513152802419,
5
+ "eval_runtime": 1.3882,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 751.321,
8
+ "eval_steps_per_second": 5.042,
9
+ "total_flos": 424333382309400.0,
10
+ "train_loss": 0.2657259335027677,
11
+ "train_runtime": 35.0333,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 732.247,
14
+ "train_steps_per_second": 9.163
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_44/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.2829234004020691,
4
+ "eval_matthews_correlation": 0.7729513152802419,
5
+ "eval_runtime": 1.3882,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 751.321,
8
+ "eval_steps_per_second": 5.042
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/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/hf_sequence_classifier/seed_44 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/seed_44/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 424333382309400.0,
4
+ "train_loss": 0.2657259335027677,
5
+ "train_runtime": 35.0333,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 732.247,
8
+ "train_steps_per_second": 9.163
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/hf_sequence_classifier/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.170408725738525,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4426,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 4.682508945465088,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.324,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.2635350525379181,
28
+ "eval_matthews_correlation": 0.7542296587798193,
29
+ "eval_runtime": 1.513,
30
+ "eval_samples_per_second": 689.353,
31
+ "eval_steps_per_second": 4.627,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 6.913776397705078,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2693,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 6.106342792510986,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2331,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2619330585002899,
51
+ "eval_matthews_correlation": 0.7824250237671008,
52
+ "eval_runtime": 1.3738,
53
+ "eval_samples_per_second": 759.182,
54
+ "eval_steps_per_second": 5.095,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 8.283414840698242,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1894,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 10.368456840515137,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1812,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.2829234004020691,
74
+ "eval_matthews_correlation": 0.7729513152802419,
75
+ "eval_runtime": 1.4952,
76
+ "eval_samples_per_second": 697.546,
77
+ "eval_steps_per_second": 4.682,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 424333382309400.0,
84
+ "train_loss": 0.2657259335027677,
85
+ "train_runtime": 35.0333,
86
+ "train_samples_per_second": 732.247,
87
+ "train_steps_per_second": 9.163
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": 424333382309400.0,
108
+ "train_batch_size": 80,
109
+ "trial_name": null,
110
+ "trial_params": null
111
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_42/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.3104800581932068,
4
+ "eval_matthews_correlation": 0.7354753979999464,
5
+ "eval_runtime": 1.5803,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 659.999,
8
+ "eval_steps_per_second": 4.43,
9
+ "total_flos": 429104574692352.0,
10
+ "train_loss": 0.26581977386712285,
11
+ "train_runtime": 29.3437,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 874.226,
14
+ "train_steps_per_second": 10.939
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_42/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.3104800581932068,
4
+ "eval_matthews_correlation": 0.7354753979999464,
5
+ "eval_runtime": 1.5803,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 659.999,
8
+ "eval_steps_per_second": 4.43
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/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 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/max/seed_42 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy max --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_42/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 429104574692352.0,
4
+ "train_loss": 0.26581977386712285,
5
+ "train_runtime": 29.3437,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 874.226,
8
+ "train_steps_per_second": 10.939
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/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.00053071975708,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4448,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 5.761870861053467,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3223,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.30257633328437805,
28
+ "eval_matthews_correlation": 0.7113064650024932,
29
+ "eval_runtime": 1.775,
30
+ "eval_samples_per_second": 587.608,
31
+ "eval_steps_per_second": 3.944,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 3.953287363052368,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2721,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 5.419213771820068,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2332,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.28807854652404785,
51
+ "eval_matthews_correlation": 0.7210288223868512,
52
+ "eval_runtime": 1.6387,
53
+ "eval_samples_per_second": 636.49,
54
+ "eval_steps_per_second": 4.272,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 6.456455230712891,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.2,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 5.79647159576416,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1687,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.3104800581932068,
74
+ "eval_matthews_correlation": 0.7354753979999464,
75
+ "eval_runtime": 1.5395,
76
+ "eval_samples_per_second": 677.48,
77
+ "eval_steps_per_second": 4.547,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 429104574692352.0,
84
+ "train_loss": 0.26581977386712285,
85
+ "train_runtime": 29.3437,
86
+ "train_samples_per_second": 874.226,
87
+ "train_steps_per_second": 10.939
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/max/seed_43/all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.2829602062702179,
4
+ "eval_matthews_correlation": 0.7638553055158107,
5
+ "eval_runtime": 1.3595,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 767.182,
8
+ "eval_steps_per_second": 5.149,
9
+ "total_flos": 426221819771904.0,
10
+ "train_loss": 0.26830103241394615,
11
+ "train_runtime": 31.2693,
12
+ "train_samples": 8551,
13
+ "train_samples_per_second": 820.39,
14
+ "train_steps_per_second": 10.266
15
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_43/eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.2829602062702179,
4
+ "eval_matthews_correlation": 0.7638553055158107,
5
+ "eval_runtime": 1.3595,
6
+ "eval_samples": 1043,
7
+ "eval_samples_per_second": 767.182,
8
+ "eval_steps_per_second": 5.149
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/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 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/max/seed_43 --overwrite_output_dir --trust_remote_code true --bf16 --tf32 true --skip_model_save --use_mha true --pooling_strategy max --n_heads 8 --head_dim 32 --classifier_hidden_dim 512 --head_dropout 0.05
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/seed_43/train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 426221819771904.0,
4
+ "train_loss": 0.26830103241394615,
5
+ "train_runtime": 31.2693,
6
+ "train_samples": 8551,
7
+ "train_samples_per_second": 820.39,
8
+ "train_steps_per_second": 10.266
9
+ }
mbert_paper_metrics/result_ablation_mbert_paper/cola/mBERT/max/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.332101345062256,
14
+ "learning_rate": 1.688473520249221e-05,
15
+ "loss": 0.4634,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.9345794392523364,
20
+ "grad_norm": 5.116937160491943,
21
+ "learning_rate": 1.3769470404984425e-05,
22
+ "loss": 0.3316,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 1.0,
27
+ "eval_loss": 0.2942867577075958,
28
+ "eval_matthews_correlation": 0.7116335016864657,
29
+ "eval_runtime": 1.6351,
30
+ "eval_samples_per_second": 637.886,
31
+ "eval_steps_per_second": 4.281,
32
+ "step": 107
33
+ },
34
+ {
35
+ "epoch": 1.4018691588785046,
36
+ "grad_norm": 4.77727746963501,
37
+ "learning_rate": 1.0654205607476635e-05,
38
+ "loss": 0.2673,
39
+ "step": 150
40
+ },
41
+ {
42
+ "epoch": 1.8691588785046729,
43
+ "grad_norm": 3.866352081298828,
44
+ "learning_rate": 7.538940809968847e-06,
45
+ "loss": 0.2258,
46
+ "step": 200
47
+ },
48
+ {
49
+ "epoch": 2.0,
50
+ "eval_loss": 0.2581494450569153,
51
+ "eval_matthews_correlation": 0.762410162302737,
52
+ "eval_runtime": 1.6224,
53
+ "eval_samples_per_second": 642.892,
54
+ "eval_steps_per_second": 4.315,
55
+ "step": 214
56
+ },
57
+ {
58
+ "epoch": 2.336448598130841,
59
+ "grad_norm": 7.299858570098877,
60
+ "learning_rate": 4.42367601246106e-06,
61
+ "loss": 0.1864,
62
+ "step": 250
63
+ },
64
+ {
65
+ "epoch": 2.803738317757009,
66
+ "grad_norm": 3.356229305267334,
67
+ "learning_rate": 1.308411214953271e-06,
68
+ "loss": 0.1754,
69
+ "step": 300
70
+ },
71
+ {
72
+ "epoch": 3.0,
73
+ "eval_loss": 0.2829602062702179,
74
+ "eval_matthews_correlation": 0.7638553055158107,
75
+ "eval_runtime": 1.328,
76
+ "eval_samples_per_second": 785.412,
77
+ "eval_steps_per_second": 5.271,
78
+ "step": 321
79
+ },
80
+ {
81
+ "epoch": 3.0,
82
+ "step": 321,
83
+ "total_flos": 426221819771904.0,
84
+ "train_loss": 0.26830103241394615,
85
+ "train_runtime": 31.2693,
86
+ "train_samples_per_second": 820.39,
87
+ "train_steps_per_second": 10.266
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
+ }