dzungpham commited on
Commit
ff9dd86
·
verified ·
1 Parent(s): e0919c7

upload unixcoder inference results

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/config.json +29 -0
  2. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/config_hyperparams.json +53 -0
  3. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/hyperparams.json +35 -0
  4. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/merges.txt +0 -0
  5. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/model.safetensors +3 -0
  6. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/optimizer.pt +3 -0
  7. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/rng_state.pth +3 -0
  8. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/scaler.pt +3 -0
  9. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/scheduler.pt +3 -0
  10. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/special_tokens_map.json +51 -0
  11. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/tokenizer.json +0 -0
  12. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/tokenizer_config.json +58 -0
  13. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/trainer_state.json +253 -0
  14. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/training_args.bin +3 -0
  15. graphcodebert-base-lowLR-highBatchSize/checkpoint-300/vocab.json +0 -0
  16. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/config.json +29 -0
  17. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/config_hyperparams.json +53 -0
  18. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/hyperparams.json +35 -0
  19. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/merges.txt +0 -0
  20. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/model.safetensors +3 -0
  21. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/optimizer.pt +3 -0
  22. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/rng_state.pth +3 -0
  23. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/scaler.pt +3 -0
  24. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/scheduler.pt +3 -0
  25. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/special_tokens_map.json +51 -0
  26. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/tokenizer.json +0 -0
  27. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/tokenizer_config.json +58 -0
  28. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/trainer_state.json +288 -0
  29. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/training_args.bin +3 -0
  30. graphcodebert-base-lowLR-highBatchSize/checkpoint-350/vocab.json +0 -0
  31. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/config.json +29 -0
  32. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/config_hyperparams.json +53 -0
  33. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/hyperparams.json +35 -0
  34. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/merges.txt +0 -0
  35. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/model.safetensors +3 -0
  36. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/optimizer.pt +3 -0
  37. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/rng_state.pth +3 -0
  38. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/scaler.pt +3 -0
  39. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/scheduler.pt +3 -0
  40. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/special_tokens_map.json +51 -0
  41. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/tokenizer.json +0 -0
  42. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/tokenizer_config.json +58 -0
  43. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/trainer_state.json +323 -0
  44. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/training_args.bin +3 -0
  45. graphcodebert-base-lowLR-highBatchSize/checkpoint-400/vocab.json +0 -0
  46. graphcodebert-base-lowLR-highBatchSize/checkpoint-450/config.json +29 -0
  47. graphcodebert-base-lowLR-highBatchSize/checkpoint-450/config_hyperparams.json +53 -0
  48. graphcodebert-base-lowLR-highBatchSize/checkpoint-450/hyperparams.json +35 -0
  49. graphcodebert-base-lowLR-highBatchSize/checkpoint-450/merges.txt +0 -0
  50. graphcodebert-base-lowLR-highBatchSize/checkpoint-450/model.safetensors +3 -0
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "RobertaForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.3,
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.3,
8
+ "dtype": "float32",
9
+ "eos_token_id": 2,
10
+ "gradient_checkpointing": false,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.3,
13
+ "hidden_size": 768,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 3072,
16
+ "layer_norm_eps": 1e-05,
17
+ "max_position_embeddings": 514,
18
+ "model_type": "roberta",
19
+ "num_attention_heads": 12,
20
+ "num_hidden_layers": 12,
21
+ "output_past": true,
22
+ "pad_token_id": 1,
23
+ "position_embedding_type": "absolute",
24
+ "problem_type": "single_label_classification",
25
+ "transformers_version": "4.56.0",
26
+ "type_vocab_size": 1,
27
+ "use_cache": true,
28
+ "vocab_size": 50265
29
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/config_hyperparams.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_config": {
3
+ "model_name": "microsoft/graphcodebert-base",
4
+ "num_epochs": 1,
5
+ "batch_size": 256,
6
+ "learning_rate": 1e-06,
7
+ "max_length": 512,
8
+ "num_labels": 2,
9
+ "loss_type": "r-drop",
10
+ "focal_alpha": 1.0,
11
+ "focal_gamma": 2.0,
12
+ "r_drop_alpha": 10.0,
13
+ "infonce_temperature": 0.07,
14
+ "infonce_weight": 0.5,
15
+ "label_smoothing": 0.5,
16
+ "adversarial_epsilon": 0.5,
17
+ "use_swa": false,
18
+ "swa_start_epoch": 0,
19
+ "swa_lr": 1e-05,
20
+ "data_augmentation": true,
21
+ "aug_rename_prob": 0.8,
22
+ "aug_format_prob": 0.8,
23
+ "freeze_base": true,
24
+ "seed": 42,
25
+ "use_wandb": true,
26
+ "mixup_alpha": 1.0,
27
+ "low_pass_keep_ratio": 0.5,
28
+ "freq_consistency_weight": 0.5
29
+ },
30
+ "training_arguments": {
31
+ "output_dir": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/",
32
+ "num_train_epochs": 1,
33
+ "per_device_train_batch_size": 256,
34
+ "per_device_eval_batch_size": 512,
35
+ "learning_rate": 1e-06,
36
+ "warmup_steps": 204,
37
+ "weight_decay": 0.1,
38
+ "logging_steps": 10,
39
+ "eval_steps": 1000,
40
+ "save_steps": 50,
41
+ "metric_for_best_model": "macro_f1",
42
+ "greater_is_better": true,
43
+ "save_total_limit": 4,
44
+ "fp16": true,
45
+ "seed": 42
46
+ },
47
+ "training_state": {
48
+ "global_step": 300,
49
+ "epoch": 0.29354207436399216,
50
+ "best_metric": null,
51
+ "best_model_checkpoint": null
52
+ }
53
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/hyperparams.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "microsoft/graphcodebert-base",
3
+ "output_dir": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/",
4
+ "num_epochs": 1,
5
+ "max_steps": -1,
6
+ "batch_size": 256,
7
+ "learning_rate": 1e-06,
8
+ "max_length": 512,
9
+ "num_labels": 2,
10
+ "use_wandb": true,
11
+ "freeze_base": true,
12
+ "loss_type": "r-drop",
13
+ "focal_alpha": 1.0,
14
+ "focal_gamma": 2.0,
15
+ "r_drop_alpha": 10.0,
16
+ "infonce_temperature": 0.07,
17
+ "infonce_weight": 0.5,
18
+ "seed": 42,
19
+ "resume_from_checkpoint": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/checkpoint-150",
20
+ "label_smoothing": 0.5,
21
+ "adversarial_epsilon": 0.5,
22
+ "use_swa": false,
23
+ "swa_start_epoch": 0,
24
+ "swa_lr": 1e-05,
25
+ "data_augmentation": true,
26
+ "aug_rename_prob": 0.8,
27
+ "aug_format_prob": 0.8,
28
+ "mixup_alpha": 1.0,
29
+ "low_pass_keep_ratio": 0.5,
30
+ "freq_consistency_weight": 0.5,
31
+ "hidden_dropout_prob": 0.3,
32
+ "attention_probs_dropout_prob": 0.3,
33
+ "classifier_dropout": 0.3,
34
+ "device": "cuda"
35
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:887ddaff8ed9f2847ee48b07006fe44accd881ddff639ed716d714b0d8cbba1e
3
+ size 498612824
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77ce20a7bc55b277f0f55837d91399a4e5305d0f417f998203c53020e9d7ba4c
3
+ size 4741859
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73aab9139f3cab35bd545916faa6eb789a3dd6684c13a4cd2232a51854fa7d3a
3
+ size 14645
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b21c5349d5e7d02de630ebc1cb53ade1d9c6079eeb8594d223bb786011a0428b
3
+ size 1383
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44e499376c1bcdb063e8e402152e4b725b246196c6753599ee18b8ce21bb2784
3
+ size 1465
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": true,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": false,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "extra_special_tokens": {},
51
+ "mask_token": "<mask>",
52
+ "model_max_length": 512,
53
+ "pad_token": "<pad>",
54
+ "sep_token": "</s>",
55
+ "tokenizer_class": "RobertaTokenizer",
56
+ "trim_offsets": true,
57
+ "unk_token": "<unk>"
58
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/trainer_state.json ADDED
@@ -0,0 +1,253 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.29354207436399216,
6
+ "eval_steps": 1000,
7
+ "global_step": 300,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.009784735812133072,
14
+ "grad_norm": 165951.1875,
15
+ "learning_rate": 4.411764705882353e-08,
16
+ "loss": 0.8416,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.019569471624266144,
21
+ "grad_norm": 206132.796875,
22
+ "learning_rate": 9.313725490196079e-08,
23
+ "loss": 0.8378,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.029354207436399216,
28
+ "grad_norm": 171197.53125,
29
+ "learning_rate": 1.4215686274509803e-07,
30
+ "loss": 0.8388,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.03913894324853229,
35
+ "grad_norm": 206661.828125,
36
+ "learning_rate": 1.9117647058823527e-07,
37
+ "loss": 0.847,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.04892367906066536,
42
+ "grad_norm": 230268.96875,
43
+ "learning_rate": 2.4019607843137255e-07,
44
+ "loss": 0.8411,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.05870841487279843,
49
+ "grad_norm": 133248.28125,
50
+ "learning_rate": 2.8921568627450984e-07,
51
+ "loss": 0.8412,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.0684931506849315,
56
+ "grad_norm": 147768.828125,
57
+ "learning_rate": 3.3823529411764707e-07,
58
+ "loss": 0.8431,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.07827788649706457,
63
+ "grad_norm": 167327.53125,
64
+ "learning_rate": 3.872549019607843e-07,
65
+ "loss": 0.8431,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 0.08806262230919765,
70
+ "grad_norm": 151943.75,
71
+ "learning_rate": 4.3627450980392154e-07,
72
+ "loss": 0.8311,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 0.09784735812133072,
77
+ "grad_norm": 149971.59375,
78
+ "learning_rate": 4.852941176470588e-07,
79
+ "loss": 0.8412,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 0.10763209393346379,
84
+ "grad_norm": 162329.40625,
85
+ "learning_rate": 5.34313725490196e-07,
86
+ "loss": 0.8356,
87
+ "step": 110
88
+ },
89
+ {
90
+ "epoch": 0.11741682974559686,
91
+ "grad_norm": 158769.765625,
92
+ "learning_rate": 5.833333333333334e-07,
93
+ "loss": 0.8351,
94
+ "step": 120
95
+ },
96
+ {
97
+ "epoch": 0.12720156555772993,
98
+ "grad_norm": 157725.59375,
99
+ "learning_rate": 6.323529411764706e-07,
100
+ "loss": 0.8331,
101
+ "step": 130
102
+ },
103
+ {
104
+ "epoch": 0.136986301369863,
105
+ "grad_norm": 205733.0,
106
+ "learning_rate": 6.813725490196079e-07,
107
+ "loss": 0.8366,
108
+ "step": 140
109
+ },
110
+ {
111
+ "epoch": 0.14677103718199608,
112
+ "grad_norm": 172581.328125,
113
+ "learning_rate": 7.30392156862745e-07,
114
+ "loss": 0.831,
115
+ "step": 150
116
+ },
117
+ {
118
+ "epoch": 0.15655577299412915,
119
+ "grad_norm": 137428.0,
120
+ "learning_rate": 7.794117647058823e-07,
121
+ "loss": 0.8326,
122
+ "step": 160
123
+ },
124
+ {
125
+ "epoch": 0.16634050880626222,
126
+ "grad_norm": 151189.203125,
127
+ "learning_rate": 8.284313725490196e-07,
128
+ "loss": 0.8281,
129
+ "step": 170
130
+ },
131
+ {
132
+ "epoch": 0.1761252446183953,
133
+ "grad_norm": 145620.625,
134
+ "learning_rate": 8.774509803921568e-07,
135
+ "loss": 0.8252,
136
+ "step": 180
137
+ },
138
+ {
139
+ "epoch": 0.18590998043052837,
140
+ "grad_norm": 169938.84375,
141
+ "learning_rate": 9.264705882352941e-07,
142
+ "loss": 0.8308,
143
+ "step": 190
144
+ },
145
+ {
146
+ "epoch": 0.19569471624266144,
147
+ "grad_norm": 138760.015625,
148
+ "learning_rate": 9.754901960784313e-07,
149
+ "loss": 0.824,
150
+ "step": 200
151
+ },
152
+ {
153
+ "epoch": 0.2054794520547945,
154
+ "grad_norm": 107872.125,
155
+ "learning_rate": 9.99631294374483e-07,
156
+ "loss": 0.8254,
157
+ "step": 210
158
+ },
159
+ {
160
+ "epoch": 0.21526418786692758,
161
+ "grad_norm": 113610.8359375,
162
+ "learning_rate": 9.966849112204936e-07,
163
+ "loss": 0.8267,
164
+ "step": 220
165
+ },
166
+ {
167
+ "epoch": 0.22504892367906065,
168
+ "grad_norm": 172441.515625,
169
+ "learning_rate": 9.908095200725339e-07,
170
+ "loss": 0.8198,
171
+ "step": 230
172
+ },
173
+ {
174
+ "epoch": 0.23483365949119372,
175
+ "grad_norm": 128079.9140625,
176
+ "learning_rate": 9.82039768787326e-07,
177
+ "loss": 0.8143,
178
+ "step": 240
179
+ },
180
+ {
181
+ "epoch": 0.2446183953033268,
182
+ "grad_norm": 98149.8984375,
183
+ "learning_rate": 9.70427373595903e-07,
184
+ "loss": 0.8191,
185
+ "step": 250
186
+ },
187
+ {
188
+ "epoch": 0.25440313111545987,
189
+ "grad_norm": 147377.4375,
190
+ "learning_rate": 9.560408141270531e-07,
191
+ "loss": 0.8178,
192
+ "step": 260
193
+ },
194
+ {
195
+ "epoch": 0.26418786692759294,
196
+ "grad_norm": 156439.40625,
197
+ "learning_rate": 9.389649295750781e-07,
198
+ "loss": 0.8085,
199
+ "step": 270
200
+ },
201
+ {
202
+ "epoch": 0.273972602739726,
203
+ "grad_norm": 131892.015625,
204
+ "learning_rate": 9.193004183933174e-07,
205
+ "loss": 0.8172,
206
+ "step": 280
207
+ },
208
+ {
209
+ "epoch": 0.2837573385518591,
210
+ "grad_norm": 132474.5,
211
+ "learning_rate": 8.971632444637971e-07,
212
+ "loss": 0.8136,
213
+ "step": 290
214
+ },
215
+ {
216
+ "epoch": 0.29354207436399216,
217
+ "grad_norm": 112619.109375,
218
+ "learning_rate": 8.726839532448928e-07,
219
+ "loss": 0.8065,
220
+ "step": 300
221
+ }
222
+ ],
223
+ "logging_steps": 10,
224
+ "max_steps": 1022,
225
+ "num_input_tokens_seen": 0,
226
+ "num_train_epochs": 1,
227
+ "save_steps": 50,
228
+ "stateful_callbacks": {
229
+ "EarlyStoppingCallback": {
230
+ "args": {
231
+ "early_stopping_patience": 3,
232
+ "early_stopping_threshold": 0.0
233
+ },
234
+ "attributes": {
235
+ "early_stopping_patience_counter": 0
236
+ }
237
+ },
238
+ "TrainerControl": {
239
+ "args": {
240
+ "should_epoch_stop": false,
241
+ "should_evaluate": false,
242
+ "should_log": false,
243
+ "should_save": true,
244
+ "should_training_stop": false
245
+ },
246
+ "attributes": {}
247
+ }
248
+ },
249
+ "total_flos": 4.0413858103296e+16,
250
+ "train_batch_size": 512,
251
+ "trial_name": null,
252
+ "trial_params": null
253
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22d0ce832e6223863ce0cb8edb6365c4f89eeb0b5c285c88ed750e293e5ffbe9
3
+ size 5905
graphcodebert-base-lowLR-highBatchSize/checkpoint-300/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "RobertaForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.3,
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.3,
8
+ "dtype": "float32",
9
+ "eos_token_id": 2,
10
+ "gradient_checkpointing": false,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.3,
13
+ "hidden_size": 768,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 3072,
16
+ "layer_norm_eps": 1e-05,
17
+ "max_position_embeddings": 514,
18
+ "model_type": "roberta",
19
+ "num_attention_heads": 12,
20
+ "num_hidden_layers": 12,
21
+ "output_past": true,
22
+ "pad_token_id": 1,
23
+ "position_embedding_type": "absolute",
24
+ "problem_type": "single_label_classification",
25
+ "transformers_version": "4.56.0",
26
+ "type_vocab_size": 1,
27
+ "use_cache": true,
28
+ "vocab_size": 50265
29
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/config_hyperparams.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_config": {
3
+ "model_name": "microsoft/graphcodebert-base",
4
+ "num_epochs": 1,
5
+ "batch_size": 256,
6
+ "learning_rate": 1e-06,
7
+ "max_length": 512,
8
+ "num_labels": 2,
9
+ "loss_type": "r-drop",
10
+ "focal_alpha": 1.0,
11
+ "focal_gamma": 2.0,
12
+ "r_drop_alpha": 10.0,
13
+ "infonce_temperature": 0.07,
14
+ "infonce_weight": 0.5,
15
+ "label_smoothing": 0.5,
16
+ "adversarial_epsilon": 0.5,
17
+ "use_swa": false,
18
+ "swa_start_epoch": 0,
19
+ "swa_lr": 1e-05,
20
+ "data_augmentation": true,
21
+ "aug_rename_prob": 0.8,
22
+ "aug_format_prob": 0.8,
23
+ "freeze_base": true,
24
+ "seed": 42,
25
+ "use_wandb": true,
26
+ "mixup_alpha": 1.0,
27
+ "low_pass_keep_ratio": 0.5,
28
+ "freq_consistency_weight": 0.5
29
+ },
30
+ "training_arguments": {
31
+ "output_dir": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/",
32
+ "num_train_epochs": 1,
33
+ "per_device_train_batch_size": 256,
34
+ "per_device_eval_batch_size": 512,
35
+ "learning_rate": 1e-06,
36
+ "warmup_steps": 204,
37
+ "weight_decay": 0.1,
38
+ "logging_steps": 10,
39
+ "eval_steps": 1000,
40
+ "save_steps": 50,
41
+ "metric_for_best_model": "macro_f1",
42
+ "greater_is_better": true,
43
+ "save_total_limit": 4,
44
+ "fp16": true,
45
+ "seed": 42
46
+ },
47
+ "training_state": {
48
+ "global_step": 350,
49
+ "epoch": 0.3424657534246575,
50
+ "best_metric": null,
51
+ "best_model_checkpoint": null
52
+ }
53
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/hyperparams.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "microsoft/graphcodebert-base",
3
+ "output_dir": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/",
4
+ "num_epochs": 1,
5
+ "max_steps": -1,
6
+ "batch_size": 256,
7
+ "learning_rate": 1e-06,
8
+ "max_length": 512,
9
+ "num_labels": 2,
10
+ "use_wandb": true,
11
+ "freeze_base": true,
12
+ "loss_type": "r-drop",
13
+ "focal_alpha": 1.0,
14
+ "focal_gamma": 2.0,
15
+ "r_drop_alpha": 10.0,
16
+ "infonce_temperature": 0.07,
17
+ "infonce_weight": 0.5,
18
+ "seed": 42,
19
+ "resume_from_checkpoint": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/checkpoint-150",
20
+ "label_smoothing": 0.5,
21
+ "adversarial_epsilon": 0.5,
22
+ "use_swa": false,
23
+ "swa_start_epoch": 0,
24
+ "swa_lr": 1e-05,
25
+ "data_augmentation": true,
26
+ "aug_rename_prob": 0.8,
27
+ "aug_format_prob": 0.8,
28
+ "mixup_alpha": 1.0,
29
+ "low_pass_keep_ratio": 0.5,
30
+ "freq_consistency_weight": 0.5,
31
+ "hidden_dropout_prob": 0.3,
32
+ "attention_probs_dropout_prob": 0.3,
33
+ "classifier_dropout": 0.3,
34
+ "device": "cuda"
35
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fef80ac3319fc044bd5ed974ce634130ad50dd3891109eedb8d138f19c9ac974
3
+ size 498612824
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc46c104c26de9b07a90c7a38f083f5e2a8d00cb7be0e639f6f74084fd942547
3
+ size 4741859
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef2287c5d715c4c2b1f2475cc59ae767683b80b8c74ca678c03ca174ce5f0e2f
3
+ size 14645
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b21c5349d5e7d02de630ebc1cb53ade1d9c6079eeb8594d223bb786011a0428b
3
+ size 1383
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:928190fa1d3edd4ba374db3d04217b214d87bc7124232d8888e8b2ae2eb12f68
3
+ size 1465
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": true,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": false,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "extra_special_tokens": {},
51
+ "mask_token": "<mask>",
52
+ "model_max_length": 512,
53
+ "pad_token": "<pad>",
54
+ "sep_token": "</s>",
55
+ "tokenizer_class": "RobertaTokenizer",
56
+ "trim_offsets": true,
57
+ "unk_token": "<unk>"
58
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/trainer_state.json ADDED
@@ -0,0 +1,288 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.3424657534246575,
6
+ "eval_steps": 1000,
7
+ "global_step": 350,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.009784735812133072,
14
+ "grad_norm": 165951.1875,
15
+ "learning_rate": 4.411764705882353e-08,
16
+ "loss": 0.8416,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.019569471624266144,
21
+ "grad_norm": 206132.796875,
22
+ "learning_rate": 9.313725490196079e-08,
23
+ "loss": 0.8378,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.029354207436399216,
28
+ "grad_norm": 171197.53125,
29
+ "learning_rate": 1.4215686274509803e-07,
30
+ "loss": 0.8388,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.03913894324853229,
35
+ "grad_norm": 206661.828125,
36
+ "learning_rate": 1.9117647058823527e-07,
37
+ "loss": 0.847,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.04892367906066536,
42
+ "grad_norm": 230268.96875,
43
+ "learning_rate": 2.4019607843137255e-07,
44
+ "loss": 0.8411,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.05870841487279843,
49
+ "grad_norm": 133248.28125,
50
+ "learning_rate": 2.8921568627450984e-07,
51
+ "loss": 0.8412,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.0684931506849315,
56
+ "grad_norm": 147768.828125,
57
+ "learning_rate": 3.3823529411764707e-07,
58
+ "loss": 0.8431,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.07827788649706457,
63
+ "grad_norm": 167327.53125,
64
+ "learning_rate": 3.872549019607843e-07,
65
+ "loss": 0.8431,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 0.08806262230919765,
70
+ "grad_norm": 151943.75,
71
+ "learning_rate": 4.3627450980392154e-07,
72
+ "loss": 0.8311,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 0.09784735812133072,
77
+ "grad_norm": 149971.59375,
78
+ "learning_rate": 4.852941176470588e-07,
79
+ "loss": 0.8412,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 0.10763209393346379,
84
+ "grad_norm": 162329.40625,
85
+ "learning_rate": 5.34313725490196e-07,
86
+ "loss": 0.8356,
87
+ "step": 110
88
+ },
89
+ {
90
+ "epoch": 0.11741682974559686,
91
+ "grad_norm": 158769.765625,
92
+ "learning_rate": 5.833333333333334e-07,
93
+ "loss": 0.8351,
94
+ "step": 120
95
+ },
96
+ {
97
+ "epoch": 0.12720156555772993,
98
+ "grad_norm": 157725.59375,
99
+ "learning_rate": 6.323529411764706e-07,
100
+ "loss": 0.8331,
101
+ "step": 130
102
+ },
103
+ {
104
+ "epoch": 0.136986301369863,
105
+ "grad_norm": 205733.0,
106
+ "learning_rate": 6.813725490196079e-07,
107
+ "loss": 0.8366,
108
+ "step": 140
109
+ },
110
+ {
111
+ "epoch": 0.14677103718199608,
112
+ "grad_norm": 172581.328125,
113
+ "learning_rate": 7.30392156862745e-07,
114
+ "loss": 0.831,
115
+ "step": 150
116
+ },
117
+ {
118
+ "epoch": 0.15655577299412915,
119
+ "grad_norm": 137428.0,
120
+ "learning_rate": 7.794117647058823e-07,
121
+ "loss": 0.8326,
122
+ "step": 160
123
+ },
124
+ {
125
+ "epoch": 0.16634050880626222,
126
+ "grad_norm": 151189.203125,
127
+ "learning_rate": 8.284313725490196e-07,
128
+ "loss": 0.8281,
129
+ "step": 170
130
+ },
131
+ {
132
+ "epoch": 0.1761252446183953,
133
+ "grad_norm": 145620.625,
134
+ "learning_rate": 8.774509803921568e-07,
135
+ "loss": 0.8252,
136
+ "step": 180
137
+ },
138
+ {
139
+ "epoch": 0.18590998043052837,
140
+ "grad_norm": 169938.84375,
141
+ "learning_rate": 9.264705882352941e-07,
142
+ "loss": 0.8308,
143
+ "step": 190
144
+ },
145
+ {
146
+ "epoch": 0.19569471624266144,
147
+ "grad_norm": 138760.015625,
148
+ "learning_rate": 9.754901960784313e-07,
149
+ "loss": 0.824,
150
+ "step": 200
151
+ },
152
+ {
153
+ "epoch": 0.2054794520547945,
154
+ "grad_norm": 107872.125,
155
+ "learning_rate": 9.99631294374483e-07,
156
+ "loss": 0.8254,
157
+ "step": 210
158
+ },
159
+ {
160
+ "epoch": 0.21526418786692758,
161
+ "grad_norm": 113610.8359375,
162
+ "learning_rate": 9.966849112204936e-07,
163
+ "loss": 0.8267,
164
+ "step": 220
165
+ },
166
+ {
167
+ "epoch": 0.22504892367906065,
168
+ "grad_norm": 172441.515625,
169
+ "learning_rate": 9.908095200725339e-07,
170
+ "loss": 0.8198,
171
+ "step": 230
172
+ },
173
+ {
174
+ "epoch": 0.23483365949119372,
175
+ "grad_norm": 128079.9140625,
176
+ "learning_rate": 9.82039768787326e-07,
177
+ "loss": 0.8143,
178
+ "step": 240
179
+ },
180
+ {
181
+ "epoch": 0.2446183953033268,
182
+ "grad_norm": 98149.8984375,
183
+ "learning_rate": 9.70427373595903e-07,
184
+ "loss": 0.8191,
185
+ "step": 250
186
+ },
187
+ {
188
+ "epoch": 0.25440313111545987,
189
+ "grad_norm": 147377.4375,
190
+ "learning_rate": 9.560408141270531e-07,
191
+ "loss": 0.8178,
192
+ "step": 260
193
+ },
194
+ {
195
+ "epoch": 0.26418786692759294,
196
+ "grad_norm": 156439.40625,
197
+ "learning_rate": 9.389649295750781e-07,
198
+ "loss": 0.8085,
199
+ "step": 270
200
+ },
201
+ {
202
+ "epoch": 0.273972602739726,
203
+ "grad_norm": 131892.015625,
204
+ "learning_rate": 9.193004183933174e-07,
205
+ "loss": 0.8172,
206
+ "step": 280
207
+ },
208
+ {
209
+ "epoch": 0.2837573385518591,
210
+ "grad_norm": 132474.5,
211
+ "learning_rate": 8.971632444637971e-07,
212
+ "loss": 0.8136,
213
+ "step": 290
214
+ },
215
+ {
216
+ "epoch": 0.29354207436399216,
217
+ "grad_norm": 112619.109375,
218
+ "learning_rate": 8.726839532448928e-07,
219
+ "loss": 0.8065,
220
+ "step": 300
221
+ },
222
+ {
223
+ "epoch": 0.30332681017612523,
224
+ "grad_norm": 94225.5703125,
225
+ "learning_rate": 8.460069019297606e-07,
226
+ "loss": 0.813,
227
+ "step": 310
228
+ },
229
+ {
230
+ "epoch": 0.3131115459882583,
231
+ "grad_norm": 117812.8203125,
232
+ "learning_rate": 8.172894081553822e-07,
233
+ "loss": 0.8022,
234
+ "step": 320
235
+ },
236
+ {
237
+ "epoch": 0.32289628180039137,
238
+ "grad_norm": 143287.125,
239
+ "learning_rate": 7.86700822282382e-07,
240
+ "loss": 0.8061,
241
+ "step": 330
242
+ },
243
+ {
244
+ "epoch": 0.33268101761252444,
245
+ "grad_norm": 107456.25,
246
+ "learning_rate": 7.544215287164917e-07,
247
+ "loss": 0.8006,
248
+ "step": 340
249
+ },
250
+ {
251
+ "epoch": 0.3424657534246575,
252
+ "grad_norm": 101521.171875,
253
+ "learning_rate": 7.206418821609812e-07,
254
+ "loss": 0.8055,
255
+ "step": 350
256
+ }
257
+ ],
258
+ "logging_steps": 10,
259
+ "max_steps": 1022,
260
+ "num_input_tokens_seen": 0,
261
+ "num_train_epochs": 1,
262
+ "save_steps": 50,
263
+ "stateful_callbacks": {
264
+ "EarlyStoppingCallback": {
265
+ "args": {
266
+ "early_stopping_patience": 3,
267
+ "early_stopping_threshold": 0.0
268
+ },
269
+ "attributes": {
270
+ "early_stopping_patience_counter": 0
271
+ }
272
+ },
273
+ "TrainerControl": {
274
+ "args": {
275
+ "should_epoch_stop": false,
276
+ "should_evaluate": false,
277
+ "should_log": false,
278
+ "should_save": true,
279
+ "should_training_stop": false
280
+ },
281
+ "attributes": {}
282
+ }
283
+ },
284
+ "total_flos": 4.7149501120512e+16,
285
+ "train_batch_size": 512,
286
+ "trial_name": null,
287
+ "trial_params": null
288
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22d0ce832e6223863ce0cb8edb6365c4f89eeb0b5c285c88ed750e293e5ffbe9
3
+ size 5905
graphcodebert-base-lowLR-highBatchSize/checkpoint-350/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "RobertaForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.3,
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.3,
8
+ "dtype": "float32",
9
+ "eos_token_id": 2,
10
+ "gradient_checkpointing": false,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.3,
13
+ "hidden_size": 768,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 3072,
16
+ "layer_norm_eps": 1e-05,
17
+ "max_position_embeddings": 514,
18
+ "model_type": "roberta",
19
+ "num_attention_heads": 12,
20
+ "num_hidden_layers": 12,
21
+ "output_past": true,
22
+ "pad_token_id": 1,
23
+ "position_embedding_type": "absolute",
24
+ "problem_type": "single_label_classification",
25
+ "transformers_version": "4.56.0",
26
+ "type_vocab_size": 1,
27
+ "use_cache": true,
28
+ "vocab_size": 50265
29
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/config_hyperparams.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_config": {
3
+ "model_name": "microsoft/graphcodebert-base",
4
+ "num_epochs": 1,
5
+ "batch_size": 256,
6
+ "learning_rate": 1e-06,
7
+ "max_length": 512,
8
+ "num_labels": 2,
9
+ "loss_type": "r-drop",
10
+ "focal_alpha": 1.0,
11
+ "focal_gamma": 2.0,
12
+ "r_drop_alpha": 10.0,
13
+ "infonce_temperature": 0.07,
14
+ "infonce_weight": 0.5,
15
+ "label_smoothing": 0.5,
16
+ "adversarial_epsilon": 0.5,
17
+ "use_swa": false,
18
+ "swa_start_epoch": 0,
19
+ "swa_lr": 1e-05,
20
+ "data_augmentation": true,
21
+ "aug_rename_prob": 0.8,
22
+ "aug_format_prob": 0.8,
23
+ "freeze_base": true,
24
+ "seed": 42,
25
+ "use_wandb": true,
26
+ "mixup_alpha": 1.0,
27
+ "low_pass_keep_ratio": 0.5,
28
+ "freq_consistency_weight": 0.5
29
+ },
30
+ "training_arguments": {
31
+ "output_dir": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/",
32
+ "num_train_epochs": 1,
33
+ "per_device_train_batch_size": 256,
34
+ "per_device_eval_batch_size": 512,
35
+ "learning_rate": 1e-06,
36
+ "warmup_steps": 204,
37
+ "weight_decay": 0.1,
38
+ "logging_steps": 10,
39
+ "eval_steps": 1000,
40
+ "save_steps": 50,
41
+ "metric_for_best_model": "macro_f1",
42
+ "greater_is_better": true,
43
+ "save_total_limit": 4,
44
+ "fp16": true,
45
+ "seed": 42
46
+ },
47
+ "training_state": {
48
+ "global_step": 400,
49
+ "epoch": 0.3913894324853229,
50
+ "best_metric": null,
51
+ "best_model_checkpoint": null
52
+ }
53
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/hyperparams.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "microsoft/graphcodebert-base",
3
+ "output_dir": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/",
4
+ "num_epochs": 1,
5
+ "max_steps": -1,
6
+ "batch_size": 256,
7
+ "learning_rate": 1e-06,
8
+ "max_length": 512,
9
+ "num_labels": 2,
10
+ "use_wandb": true,
11
+ "freeze_base": true,
12
+ "loss_type": "r-drop",
13
+ "focal_alpha": 1.0,
14
+ "focal_gamma": 2.0,
15
+ "r_drop_alpha": 10.0,
16
+ "infonce_temperature": 0.07,
17
+ "infonce_weight": 0.5,
18
+ "seed": 42,
19
+ "resume_from_checkpoint": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/checkpoint-150",
20
+ "label_smoothing": 0.5,
21
+ "adversarial_epsilon": 0.5,
22
+ "use_swa": false,
23
+ "swa_start_epoch": 0,
24
+ "swa_lr": 1e-05,
25
+ "data_augmentation": true,
26
+ "aug_rename_prob": 0.8,
27
+ "aug_format_prob": 0.8,
28
+ "mixup_alpha": 1.0,
29
+ "low_pass_keep_ratio": 0.5,
30
+ "freq_consistency_weight": 0.5,
31
+ "hidden_dropout_prob": 0.3,
32
+ "attention_probs_dropout_prob": 0.3,
33
+ "classifier_dropout": 0.3,
34
+ "device": "cuda"
35
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ee10a85c7345f840ced5553e4c296702832c3b8bc1ad78d1524d2e1e0581559
3
+ size 498612824
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:098d03c86fe53a3089a267670691e3962e4c472851441ae635a00cc4591c4859
3
+ size 4741859
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68a71ce2adaa088d39a621c53df9b35a02f7ea78037e7f34b85f7d46bbafa926
3
+ size 14645
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b21c5349d5e7d02de630ebc1cb53ade1d9c6079eeb8594d223bb786011a0428b
3
+ size 1383
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:751709a16f55bce9f670805512a4ac28a4b3f402b1eecc52a1097cfeac6fb5d7
3
+ size 1465
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": true,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": false,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "extra_special_tokens": {},
51
+ "mask_token": "<mask>",
52
+ "model_max_length": 512,
53
+ "pad_token": "<pad>",
54
+ "sep_token": "</s>",
55
+ "tokenizer_class": "RobertaTokenizer",
56
+ "trim_offsets": true,
57
+ "unk_token": "<unk>"
58
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/trainer_state.json ADDED
@@ -0,0 +1,323 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.3913894324853229,
6
+ "eval_steps": 1000,
7
+ "global_step": 400,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.009784735812133072,
14
+ "grad_norm": 165951.1875,
15
+ "learning_rate": 4.411764705882353e-08,
16
+ "loss": 0.8416,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.019569471624266144,
21
+ "grad_norm": 206132.796875,
22
+ "learning_rate": 9.313725490196079e-08,
23
+ "loss": 0.8378,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.029354207436399216,
28
+ "grad_norm": 171197.53125,
29
+ "learning_rate": 1.4215686274509803e-07,
30
+ "loss": 0.8388,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.03913894324853229,
35
+ "grad_norm": 206661.828125,
36
+ "learning_rate": 1.9117647058823527e-07,
37
+ "loss": 0.847,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.04892367906066536,
42
+ "grad_norm": 230268.96875,
43
+ "learning_rate": 2.4019607843137255e-07,
44
+ "loss": 0.8411,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.05870841487279843,
49
+ "grad_norm": 133248.28125,
50
+ "learning_rate": 2.8921568627450984e-07,
51
+ "loss": 0.8412,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.0684931506849315,
56
+ "grad_norm": 147768.828125,
57
+ "learning_rate": 3.3823529411764707e-07,
58
+ "loss": 0.8431,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.07827788649706457,
63
+ "grad_norm": 167327.53125,
64
+ "learning_rate": 3.872549019607843e-07,
65
+ "loss": 0.8431,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 0.08806262230919765,
70
+ "grad_norm": 151943.75,
71
+ "learning_rate": 4.3627450980392154e-07,
72
+ "loss": 0.8311,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 0.09784735812133072,
77
+ "grad_norm": 149971.59375,
78
+ "learning_rate": 4.852941176470588e-07,
79
+ "loss": 0.8412,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 0.10763209393346379,
84
+ "grad_norm": 162329.40625,
85
+ "learning_rate": 5.34313725490196e-07,
86
+ "loss": 0.8356,
87
+ "step": 110
88
+ },
89
+ {
90
+ "epoch": 0.11741682974559686,
91
+ "grad_norm": 158769.765625,
92
+ "learning_rate": 5.833333333333334e-07,
93
+ "loss": 0.8351,
94
+ "step": 120
95
+ },
96
+ {
97
+ "epoch": 0.12720156555772993,
98
+ "grad_norm": 157725.59375,
99
+ "learning_rate": 6.323529411764706e-07,
100
+ "loss": 0.8331,
101
+ "step": 130
102
+ },
103
+ {
104
+ "epoch": 0.136986301369863,
105
+ "grad_norm": 205733.0,
106
+ "learning_rate": 6.813725490196079e-07,
107
+ "loss": 0.8366,
108
+ "step": 140
109
+ },
110
+ {
111
+ "epoch": 0.14677103718199608,
112
+ "grad_norm": 172581.328125,
113
+ "learning_rate": 7.30392156862745e-07,
114
+ "loss": 0.831,
115
+ "step": 150
116
+ },
117
+ {
118
+ "epoch": 0.15655577299412915,
119
+ "grad_norm": 137428.0,
120
+ "learning_rate": 7.794117647058823e-07,
121
+ "loss": 0.8326,
122
+ "step": 160
123
+ },
124
+ {
125
+ "epoch": 0.16634050880626222,
126
+ "grad_norm": 151189.203125,
127
+ "learning_rate": 8.284313725490196e-07,
128
+ "loss": 0.8281,
129
+ "step": 170
130
+ },
131
+ {
132
+ "epoch": 0.1761252446183953,
133
+ "grad_norm": 145620.625,
134
+ "learning_rate": 8.774509803921568e-07,
135
+ "loss": 0.8252,
136
+ "step": 180
137
+ },
138
+ {
139
+ "epoch": 0.18590998043052837,
140
+ "grad_norm": 169938.84375,
141
+ "learning_rate": 9.264705882352941e-07,
142
+ "loss": 0.8308,
143
+ "step": 190
144
+ },
145
+ {
146
+ "epoch": 0.19569471624266144,
147
+ "grad_norm": 138760.015625,
148
+ "learning_rate": 9.754901960784313e-07,
149
+ "loss": 0.824,
150
+ "step": 200
151
+ },
152
+ {
153
+ "epoch": 0.2054794520547945,
154
+ "grad_norm": 107872.125,
155
+ "learning_rate": 9.99631294374483e-07,
156
+ "loss": 0.8254,
157
+ "step": 210
158
+ },
159
+ {
160
+ "epoch": 0.21526418786692758,
161
+ "grad_norm": 113610.8359375,
162
+ "learning_rate": 9.966849112204936e-07,
163
+ "loss": 0.8267,
164
+ "step": 220
165
+ },
166
+ {
167
+ "epoch": 0.22504892367906065,
168
+ "grad_norm": 172441.515625,
169
+ "learning_rate": 9.908095200725339e-07,
170
+ "loss": 0.8198,
171
+ "step": 230
172
+ },
173
+ {
174
+ "epoch": 0.23483365949119372,
175
+ "grad_norm": 128079.9140625,
176
+ "learning_rate": 9.82039768787326e-07,
177
+ "loss": 0.8143,
178
+ "step": 240
179
+ },
180
+ {
181
+ "epoch": 0.2446183953033268,
182
+ "grad_norm": 98149.8984375,
183
+ "learning_rate": 9.70427373595903e-07,
184
+ "loss": 0.8191,
185
+ "step": 250
186
+ },
187
+ {
188
+ "epoch": 0.25440313111545987,
189
+ "grad_norm": 147377.4375,
190
+ "learning_rate": 9.560408141270531e-07,
191
+ "loss": 0.8178,
192
+ "step": 260
193
+ },
194
+ {
195
+ "epoch": 0.26418786692759294,
196
+ "grad_norm": 156439.40625,
197
+ "learning_rate": 9.389649295750781e-07,
198
+ "loss": 0.8085,
199
+ "step": 270
200
+ },
201
+ {
202
+ "epoch": 0.273972602739726,
203
+ "grad_norm": 131892.015625,
204
+ "learning_rate": 9.193004183933174e-07,
205
+ "loss": 0.8172,
206
+ "step": 280
207
+ },
208
+ {
209
+ "epoch": 0.2837573385518591,
210
+ "grad_norm": 132474.5,
211
+ "learning_rate": 8.971632444637971e-07,
212
+ "loss": 0.8136,
213
+ "step": 290
214
+ },
215
+ {
216
+ "epoch": 0.29354207436399216,
217
+ "grad_norm": 112619.109375,
218
+ "learning_rate": 8.726839532448928e-07,
219
+ "loss": 0.8065,
220
+ "step": 300
221
+ },
222
+ {
223
+ "epoch": 0.30332681017612523,
224
+ "grad_norm": 94225.5703125,
225
+ "learning_rate": 8.460069019297606e-07,
226
+ "loss": 0.813,
227
+ "step": 310
228
+ },
229
+ {
230
+ "epoch": 0.3131115459882583,
231
+ "grad_norm": 117812.8203125,
232
+ "learning_rate": 8.172894081553822e-07,
233
+ "loss": 0.8022,
234
+ "step": 320
235
+ },
236
+ {
237
+ "epoch": 0.32289628180039137,
238
+ "grad_norm": 143287.125,
239
+ "learning_rate": 7.86700822282382e-07,
240
+ "loss": 0.8061,
241
+ "step": 330
242
+ },
243
+ {
244
+ "epoch": 0.33268101761252444,
245
+ "grad_norm": 107456.25,
246
+ "learning_rate": 7.544215287164917e-07,
247
+ "loss": 0.8006,
248
+ "step": 340
249
+ },
250
+ {
251
+ "epoch": 0.3424657534246575,
252
+ "grad_norm": 101521.171875,
253
+ "learning_rate": 7.206418821609812e-07,
254
+ "loss": 0.8055,
255
+ "step": 350
256
+ },
257
+ {
258
+ "epoch": 0.3522504892367906,
259
+ "grad_norm": 118311.5234375,
260
+ "learning_rate": 6.855610850731017e-07,
261
+ "loss": 0.8081,
262
+ "step": 360
263
+ },
264
+ {
265
+ "epoch": 0.36203522504892366,
266
+ "grad_norm": 89621.828125,
267
+ "learning_rate": 6.493860129443046e-07,
268
+ "loss": 0.8058,
269
+ "step": 370
270
+ },
271
+ {
272
+ "epoch": 0.37181996086105673,
273
+ "grad_norm": 143168.65625,
274
+ "learning_rate": 6.123299943316975e-07,
275
+ "loss": 0.8028,
276
+ "step": 380
277
+ },
278
+ {
279
+ "epoch": 0.3816046966731898,
280
+ "grad_norm": 118598.2890625,
281
+ "learning_rate": 5.746115528350295e-07,
282
+ "loss": 0.8032,
283
+ "step": 390
284
+ },
285
+ {
286
+ "epoch": 0.3913894324853229,
287
+ "grad_norm": 111807.8203125,
288
+ "learning_rate": 5.364531184379139e-07,
289
+ "loss": 0.8014,
290
+ "step": 400
291
+ }
292
+ ],
293
+ "logging_steps": 10,
294
+ "max_steps": 1022,
295
+ "num_input_tokens_seen": 0,
296
+ "num_train_epochs": 1,
297
+ "save_steps": 50,
298
+ "stateful_callbacks": {
299
+ "EarlyStoppingCallback": {
300
+ "args": {
301
+ "early_stopping_patience": 3,
302
+ "early_stopping_threshold": 0.0
303
+ },
304
+ "attributes": {
305
+ "early_stopping_patience_counter": 0
306
+ }
307
+ },
308
+ "TrainerControl": {
309
+ "args": {
310
+ "should_epoch_stop": false,
311
+ "should_evaluate": false,
312
+ "should_log": false,
313
+ "should_save": true,
314
+ "should_training_stop": false
315
+ },
316
+ "attributes": {}
317
+ }
318
+ },
319
+ "total_flos": 5.3885144137728e+16,
320
+ "train_batch_size": 512,
321
+ "trial_name": null,
322
+ "trial_params": null
323
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22d0ce832e6223863ce0cb8edb6365c4f89eeb0b5c285c88ed750e293e5ffbe9
3
+ size 5905
graphcodebert-base-lowLR-highBatchSize/checkpoint-400/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-450/config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "RobertaForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.3,
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.3,
8
+ "dtype": "float32",
9
+ "eos_token_id": 2,
10
+ "gradient_checkpointing": false,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.3,
13
+ "hidden_size": 768,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 3072,
16
+ "layer_norm_eps": 1e-05,
17
+ "max_position_embeddings": 514,
18
+ "model_type": "roberta",
19
+ "num_attention_heads": 12,
20
+ "num_hidden_layers": 12,
21
+ "output_past": true,
22
+ "pad_token_id": 1,
23
+ "position_embedding_type": "absolute",
24
+ "problem_type": "single_label_classification",
25
+ "transformers_version": "4.56.0",
26
+ "type_vocab_size": 1,
27
+ "use_cache": true,
28
+ "vocab_size": 50265
29
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-450/config_hyperparams.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_config": {
3
+ "model_name": "microsoft/graphcodebert-base",
4
+ "num_epochs": 1,
5
+ "batch_size": 256,
6
+ "learning_rate": 1e-06,
7
+ "max_length": 512,
8
+ "num_labels": 2,
9
+ "loss_type": "r-drop",
10
+ "focal_alpha": 1.0,
11
+ "focal_gamma": 2.0,
12
+ "r_drop_alpha": 10.0,
13
+ "infonce_temperature": 0.07,
14
+ "infonce_weight": 0.5,
15
+ "label_smoothing": 0.5,
16
+ "adversarial_epsilon": 0.5,
17
+ "use_swa": false,
18
+ "swa_start_epoch": 0,
19
+ "swa_lr": 1e-05,
20
+ "data_augmentation": true,
21
+ "aug_rename_prob": 0.8,
22
+ "aug_format_prob": 0.8,
23
+ "freeze_base": true,
24
+ "seed": 42,
25
+ "use_wandb": true,
26
+ "mixup_alpha": 1.0,
27
+ "low_pass_keep_ratio": 0.5,
28
+ "freq_consistency_weight": 0.5
29
+ },
30
+ "training_arguments": {
31
+ "output_dir": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/",
32
+ "num_train_epochs": 1,
33
+ "per_device_train_batch_size": 256,
34
+ "per_device_eval_batch_size": 512,
35
+ "learning_rate": 1e-06,
36
+ "warmup_steps": 204,
37
+ "weight_decay": 0.1,
38
+ "logging_steps": 10,
39
+ "eval_steps": 1000,
40
+ "save_steps": 50,
41
+ "metric_for_best_model": "macro_f1",
42
+ "greater_is_better": true,
43
+ "save_total_limit": 4,
44
+ "fp16": true,
45
+ "seed": 42
46
+ },
47
+ "training_state": {
48
+ "global_step": 450,
49
+ "epoch": 0.44031311154598823,
50
+ "best_metric": null,
51
+ "best_model_checkpoint": null
52
+ }
53
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-450/hyperparams.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "microsoft/graphcodebert-base",
3
+ "output_dir": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/",
4
+ "num_epochs": 1,
5
+ "max_steps": -1,
6
+ "batch_size": 256,
7
+ "learning_rate": 1e-06,
8
+ "max_length": 512,
9
+ "num_labels": 2,
10
+ "use_wandb": true,
11
+ "freeze_base": true,
12
+ "loss_type": "r-drop",
13
+ "focal_alpha": 1.0,
14
+ "focal_gamma": 2.0,
15
+ "r_drop_alpha": 10.0,
16
+ "infonce_temperature": 0.07,
17
+ "infonce_weight": 0.5,
18
+ "seed": 42,
19
+ "resume_from_checkpoint": "output_checkpoints/graphcodebert-base-lowLR-highBatchSize/checkpoint-150",
20
+ "label_smoothing": 0.5,
21
+ "adversarial_epsilon": 0.5,
22
+ "use_swa": false,
23
+ "swa_start_epoch": 0,
24
+ "swa_lr": 1e-05,
25
+ "data_augmentation": true,
26
+ "aug_rename_prob": 0.8,
27
+ "aug_format_prob": 0.8,
28
+ "mixup_alpha": 1.0,
29
+ "low_pass_keep_ratio": 0.5,
30
+ "freq_consistency_weight": 0.5,
31
+ "hidden_dropout_prob": 0.3,
32
+ "attention_probs_dropout_prob": 0.3,
33
+ "classifier_dropout": 0.3,
34
+ "device": "cuda"
35
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-450/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-450/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3547af00f99349378f21a83a224cddcbe0bc25463fe209932447cdb2f07a1a4b
3
+ size 498612824