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

upload checkpoints best at 500

Browse files
Files changed (31) hide show
  1. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/config.json +29 -0
  2. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/config_hyperparams.json +53 -0
  3. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/hyperparams.json +35 -0
  4. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/merges.txt +0 -0
  5. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/model.safetensors +3 -0
  6. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/optimizer.pt +3 -0
  7. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/rng_state.pth +3 -0
  8. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/scaler.pt +3 -0
  9. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/scheduler.pt +3 -0
  10. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/special_tokens_map.json +51 -0
  11. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/tokenizer.json +0 -0
  12. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/tokenizer_config.json +58 -0
  13. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/trainer_state.json +393 -0
  14. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/training_args.bin +3 -0
  15. graphcodebert-base-lowLR-highBatchSize/checkpoint-500/vocab.json +0 -0
  16. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/config.json +29 -0
  17. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/config_hyperparams.json +53 -0
  18. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/hyperparams.json +35 -0
  19. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/merges.txt +0 -0
  20. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/model.safetensors +3 -0
  21. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/optimizer.pt +3 -0
  22. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/rng_state.pth +3 -0
  23. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/scaler.pt +3 -0
  24. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/scheduler.pt +3 -0
  25. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/special_tokens_map.json +51 -0
  26. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/tokenizer.json +0 -0
  27. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/tokenizer_config.json +58 -0
  28. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/trainer_state.json +428 -0
  29. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/training_args.bin +3 -0
  30. graphcodebert-base-lowLR-highBatchSize/checkpoint-550/vocab.json +0 -0
  31. graphcodebert-base-lowLR-highBatchSize/training.log +46 -46
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/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-500/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": 500,
49
+ "epoch": 0.4892367906066536,
50
+ "best_metric": null,
51
+ "best_model_checkpoint": null
52
+ }
53
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/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-450",
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-500/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62afd78fff103664d60d34ba3d06f2e7b451350dbc1f5f43dca6b0c42a813f0a
3
+ size 498612824
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9bd2ec8c5a16b46ce4afe1dcac55b713136ceebbe600c657923d4c518562f65d
3
+ size 4741859
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d0f6ebf28238c654ab9e878a40e4b165e8986b10ff2c98367e9d4fe7118c974
3
+ size 14645
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/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-500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf4e05d599e3f7350270baeea47a97754f2c1df2623cb7794b38f920e23d6222
3
+ size 1465
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/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-500/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/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-500/trainer_state.json ADDED
@@ -0,0 +1,393 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.4892367906066536,
6
+ "eval_steps": 1000,
7
+ "global_step": 500,
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
+ "epoch": 0.40117416829745595,
294
+ "grad_norm": 124106.1328125,
295
+ "learning_rate": 4.980797158126566e-07,
296
+ "loss": 0.8045,
297
+ "step": 410
298
+ },
299
+ {
300
+ "epoch": 0.410958904109589,
301
+ "grad_norm": 102746.4765625,
302
+ "learning_rate": 4.597176373239061e-07,
303
+ "loss": 0.7996,
304
+ "step": 420
305
+ },
306
+ {
307
+ "epoch": 0.4207436399217221,
308
+ "grad_norm": 108317.5859375,
309
+ "learning_rate": 4.215931085565761e-07,
310
+ "loss": 0.8037,
311
+ "step": 430
312
+ },
313
+ {
314
+ "epoch": 0.43052837573385516,
315
+ "grad_norm": 113342.515625,
316
+ "learning_rate": 3.8393095423757764e-07,
317
+ "loss": 0.8047,
318
+ "step": 440
319
+ },
320
+ {
321
+ "epoch": 0.44031311154598823,
322
+ "grad_norm": 111908.90625,
323
+ "learning_rate": 3.469532724185699e-07,
324
+ "loss": 0.8021,
325
+ "step": 450
326
+ },
327
+ {
328
+ "epoch": 0.4500978473581213,
329
+ "grad_norm": 85004.15625,
330
+ "learning_rate": 3.1087812473822845e-07,
331
+ "loss": 0.8012,
332
+ "step": 460
333
+ },
334
+ {
335
+ "epoch": 0.4598825831702544,
336
+ "grad_norm": 98335.1796875,
337
+ "learning_rate": 2.7591825048770643e-07,
338
+ "loss": 0.7977,
339
+ "step": 470
340
+ },
341
+ {
342
+ "epoch": 0.46966731898238745,
343
+ "grad_norm": 110674.03125,
344
+ "learning_rate": 2.422798120625886e-07,
345
+ "loss": 0.7979,
346
+ "step": 480
347
+ },
348
+ {
349
+ "epoch": 0.4794520547945205,
350
+ "grad_norm": 114861.5390625,
351
+ "learning_rate": 2.1016117919955512e-07,
352
+ "loss": 0.8014,
353
+ "step": 490
354
+ },
355
+ {
356
+ "epoch": 0.4892367906066536,
357
+ "grad_norm": 130109.5703125,
358
+ "learning_rate": 1.7975175916725031e-07,
359
+ "loss": 0.7994,
360
+ "step": 500
361
+ }
362
+ ],
363
+ "logging_steps": 10,
364
+ "max_steps": 1022,
365
+ "num_input_tokens_seen": 0,
366
+ "num_train_epochs": 1,
367
+ "save_steps": 50,
368
+ "stateful_callbacks": {
369
+ "EarlyStoppingCallback": {
370
+ "args": {
371
+ "early_stopping_patience": 3,
372
+ "early_stopping_threshold": 0.0
373
+ },
374
+ "attributes": {
375
+ "early_stopping_patience_counter": 0
376
+ }
377
+ },
378
+ "TrainerControl": {
379
+ "args": {
380
+ "should_epoch_stop": false,
381
+ "should_evaluate": false,
382
+ "should_log": false,
383
+ "should_save": true,
384
+ "should_training_stop": false
385
+ },
386
+ "attributes": {}
387
+ }
388
+ },
389
+ "total_flos": 6.735643017216e+16,
390
+ "train_batch_size": 512,
391
+ "trial_name": null,
392
+ "trial_params": null
393
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:029b4a858ef9e29cdaae98df63ce69dafcd44e3b7d6a390188782ba973f6387f
3
+ size 5905
graphcodebert-base-lowLR-highBatchSize/checkpoint-500/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/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-550/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": 550,
49
+ "epoch": 0.538160469667319,
50
+ "best_metric": null,
51
+ "best_model_checkpoint": null
52
+ }
53
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/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-500",
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-550/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3522ebeb3e76886d055b63e7dbb614e25d4eb7eaa31fe407b0f6ce724a359f2f
3
+ size 498612824
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ae0d8c7eb0dc55da7708c72806401e00399ec15bc9baffb93e7d1dd7df1855b
3
+ size 4741859
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:474796bdfb8d46c532dcad5c0ea3578a64f5414e0a71f978bc8ddf579ad2fe06
3
+ size 14645
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/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-550/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ca58c7fdf81d7990567985713093efa517628c10ca166dff7d9b633a6876e26
3
+ size 1465
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/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-550/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/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-550/trainer_state.json ADDED
@@ -0,0 +1,428 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.538160469667319,
6
+ "eval_steps": 1000,
7
+ "global_step": 550,
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
+ "epoch": 0.40117416829745595,
294
+ "grad_norm": 124106.1328125,
295
+ "learning_rate": 4.980797158126566e-07,
296
+ "loss": 0.8045,
297
+ "step": 410
298
+ },
299
+ {
300
+ "epoch": 0.410958904109589,
301
+ "grad_norm": 102746.4765625,
302
+ "learning_rate": 4.597176373239061e-07,
303
+ "loss": 0.7996,
304
+ "step": 420
305
+ },
306
+ {
307
+ "epoch": 0.4207436399217221,
308
+ "grad_norm": 108317.5859375,
309
+ "learning_rate": 4.215931085565761e-07,
310
+ "loss": 0.8037,
311
+ "step": 430
312
+ },
313
+ {
314
+ "epoch": 0.43052837573385516,
315
+ "grad_norm": 113342.515625,
316
+ "learning_rate": 3.8393095423757764e-07,
317
+ "loss": 0.8047,
318
+ "step": 440
319
+ },
320
+ {
321
+ "epoch": 0.44031311154598823,
322
+ "grad_norm": 111908.90625,
323
+ "learning_rate": 3.469532724185699e-07,
324
+ "loss": 0.8021,
325
+ "step": 450
326
+ },
327
+ {
328
+ "epoch": 0.4500978473581213,
329
+ "grad_norm": 85004.15625,
330
+ "learning_rate": 3.1087812473822845e-07,
331
+ "loss": 0.8012,
332
+ "step": 460
333
+ },
334
+ {
335
+ "epoch": 0.4598825831702544,
336
+ "grad_norm": 98335.1796875,
337
+ "learning_rate": 2.7591825048770643e-07,
338
+ "loss": 0.7977,
339
+ "step": 470
340
+ },
341
+ {
342
+ "epoch": 0.46966731898238745,
343
+ "grad_norm": 110674.03125,
344
+ "learning_rate": 2.422798120625886e-07,
345
+ "loss": 0.7979,
346
+ "step": 480
347
+ },
348
+ {
349
+ "epoch": 0.4794520547945205,
350
+ "grad_norm": 114861.5390625,
351
+ "learning_rate": 2.1016117919955512e-07,
352
+ "loss": 0.8014,
353
+ "step": 490
354
+ },
355
+ {
356
+ "epoch": 0.4892367906066536,
357
+ "grad_norm": 130109.5703125,
358
+ "learning_rate": 1.7975175916725031e-07,
359
+ "loss": 0.7994,
360
+ "step": 500
361
+ },
362
+ {
363
+ "epoch": 0.49902152641878667,
364
+ "grad_norm": 100678.8671875,
365
+ "learning_rate": 1.5123087980985606e-07,
366
+ "loss": 0.802,
367
+ "step": 510
368
+ },
369
+ {
370
+ "epoch": 0.5088062622309197,
371
+ "grad_norm": 115718.078125,
372
+ "learning_rate": 1.2476673203018728e-07,
373
+ "loss": 0.7971,
374
+ "step": 520
375
+ },
376
+ {
377
+ "epoch": 0.5185909980430529,
378
+ "grad_norm": 94845.9453125,
379
+ "learning_rate": 1.0051537794861287e-07,
380
+ "loss": 0.7952,
381
+ "step": 530
382
+ },
383
+ {
384
+ "epoch": 0.5283757338551859,
385
+ "grad_norm": 106283.5703125,
386
+ "learning_rate": 7.861983058679871e-08,
387
+ "loss": 0.8013,
388
+ "step": 540
389
+ },
390
+ {
391
+ "epoch": 0.538160469667319,
392
+ "grad_norm": 93847.1640625,
393
+ "learning_rate": 5.920921050348626e-08,
394
+ "loss": 0.7968,
395
+ "step": 550
396
+ }
397
+ ],
398
+ "logging_steps": 10,
399
+ "max_steps": 1022,
400
+ "num_input_tokens_seen": 0,
401
+ "num_train_epochs": 1,
402
+ "save_steps": 50,
403
+ "stateful_callbacks": {
404
+ "EarlyStoppingCallback": {
405
+ "args": {
406
+ "early_stopping_patience": 3,
407
+ "early_stopping_threshold": 0.0
408
+ },
409
+ "attributes": {
410
+ "early_stopping_patience_counter": 0
411
+ }
412
+ },
413
+ "TrainerControl": {
414
+ "args": {
415
+ "should_epoch_stop": false,
416
+ "should_evaluate": false,
417
+ "should_log": false,
418
+ "should_save": true,
419
+ "should_training_stop": false
420
+ },
421
+ "attributes": {}
422
+ }
423
+ },
424
+ "total_flos": 7.4092073189376e+16,
425
+ "train_batch_size": 512,
426
+ "trial_name": null,
427
+ "trial_params": null
428
+ }
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ede5769c73b6c610c4628bbd74325244be057dfbd9cce16276706ada4cb7f175
3
+ size 5905
graphcodebert-base-lowLR-highBatchSize/checkpoint-550/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
graphcodebert-base-lowLR-highBatchSize/training.log CHANGED
@@ -1,40 +1,40 @@
1
- 2026-04-24 15:00:13,875 - INFO - ===== Training Configuration =====
2
- 2026-04-24 15:00:13,877 - INFO - model_name : microsoft/graphcodebert-base
3
- 2026-04-24 15:00:13,878 - INFO - output_dir : output_checkpoints/graphcodebert-base-lowLR-highBatchSize/
4
- 2026-04-24 15:00:13,880 - INFO - num_epochs : 1
5
- 2026-04-24 15:00:13,882 - INFO - max_steps : -1
6
- 2026-04-24 15:00:13,883 - INFO - batch_size : 256
7
- 2026-04-24 15:00:13,884 - INFO - learning_rate : 1e-06
8
- 2026-04-24 15:00:13,885 - INFO - max_length : 512
9
- 2026-04-24 15:00:13,887 - INFO - num_labels : 2
10
- 2026-04-24 15:00:13,889 - INFO - use_wandb : True
11
- 2026-04-24 15:00:13,890 - INFO - freeze_base : True
12
- 2026-04-24 15:00:13,891 - INFO - loss_type : r-drop
13
- 2026-04-24 15:00:13,893 - INFO - focal_alpha : 1.0
14
- 2026-04-24 15:00:13,895 - INFO - focal_gamma : 2.0
15
- 2026-04-24 15:00:13,896 - INFO - r_drop_alpha : 10.0
16
- 2026-04-24 15:00:13,897 - INFO - infonce_temperature : 0.07
17
- 2026-04-24 15:00:13,899 - INFO - infonce_weight : 0.5
18
- 2026-04-24 15:00:13,900 - INFO - seed : 42
19
- 2026-04-24 15:00:13,901 - INFO - resume_from_checkpoint : output_checkpoints/graphcodebert-base-lowLR-highBatchSize/checkpoint-150
20
- 2026-04-24 15:00:13,903 - INFO - label_smoothing : 0.5
21
- 2026-04-24 15:00:13,904 - INFO - adversarial_epsilon : 0.5
22
- 2026-04-24 15:00:13,906 - INFO - use_swa : False
23
- 2026-04-24 15:00:13,907 - INFO - swa_start_epoch : 0
24
- 2026-04-24 15:00:13,909 - INFO - swa_lr : 1e-05
25
- 2026-04-24 15:00:13,910 - INFO - data_augmentation : True
26
- 2026-04-24 15:00:13,912 - INFO - aug_rename_prob : 0.8
27
- 2026-04-24 15:00:13,913 - INFO - aug_format_prob : 0.8
28
- 2026-04-24 15:00:13,914 - INFO - mixup_alpha : 1.0
29
- 2026-04-24 15:00:13,916 - INFO - low_pass_keep_ratio : 0.5
30
- 2026-04-24 15:00:13,918 - INFO - freq_consistency_weight : 0.5
31
- 2026-04-24 15:00:13,920 - INFO - hidden_dropout_prob : 0.3
32
- 2026-04-24 15:00:13,921 - INFO - attention_probs_dropout_prob : 0.3
33
- 2026-04-24 15:00:13,922 - INFO - classifier_dropout : 0.3
34
- 2026-04-24 15:00:13,924 - INFO - =================================
35
- 2026-04-24 15:00:15,022 - INFO - Model placed on cuda
36
- 2026-04-24 15:00:15,027 - INFO - ===== Model Architecture =====
37
- 2026-04-24 15:00:15,029 - INFO -
38
  RobertaForSequenceClassification(
39
  (roberta): RobertaModel(
40
  (embeddings): RobertaEmbeddings(
@@ -79,12 +79,12 @@ RobertaForSequenceClassification(
79
  (out_proj): Linear(in_features=768, out_features=2, bias=True)
80
  )
81
  )
82
- 2026-04-24 15:00:15,032 - INFO - ===== Parameter Summary =====
83
- 2026-04-24 15:00:15,033 - INFO - Total Parameters: 124,647,170
84
- 2026-04-24 15:00:15,035 - INFO - Trainable Parameters: 592,130
85
- 2026-04-24 15:00:15,036 - INFO - Non-trainable Parameters: 124,055,040
86
- 2026-04-24 15:00:15,037 - INFO - ===== Tokenizer Summary =====
87
- 2026-04-24 15:00:15,053 - INFO - Vocab size: 50265 | Special tokens: ['<s>', '</s>', '<unk>', '<pad>', '<mask>']
88
- 2026-04-24 15:00:15,054 - INFO - ===== End of Architecture Log =====
89
- 2026-04-24 15:00:15,056 - INFO - Data augmentation enabled (rename=0.8, format=0.8)
90
- 2026-04-24 15:00:50,114 - INFO - === Starting training with MixCode + FFT low-pass consistency ===
 
1
+ 2026-04-24 16:49:08,923 - INFO - ===== Training Configuration =====
2
+ 2026-04-24 16:49:08,924 - INFO - model_name : microsoft/graphcodebert-base
3
+ 2026-04-24 16:49:08,924 - INFO - output_dir : output_checkpoints/graphcodebert-base-lowLR-highBatchSize/
4
+ 2026-04-24 16:49:08,925 - INFO - num_epochs : 1
5
+ 2026-04-24 16:49:08,926 - INFO - max_steps : -1
6
+ 2026-04-24 16:49:08,927 - INFO - batch_size : 256
7
+ 2026-04-24 16:49:08,928 - INFO - learning_rate : 1e-06
8
+ 2026-04-24 16:49:08,929 - INFO - max_length : 512
9
+ 2026-04-24 16:49:08,930 - INFO - num_labels : 2
10
+ 2026-04-24 16:49:08,931 - INFO - use_wandb : True
11
+ 2026-04-24 16:49:08,931 - INFO - freeze_base : True
12
+ 2026-04-24 16:49:08,932 - INFO - loss_type : r-drop
13
+ 2026-04-24 16:49:08,933 - INFO - focal_alpha : 1.0
14
+ 2026-04-24 16:49:08,934 - INFO - focal_gamma : 2.0
15
+ 2026-04-24 16:49:08,934 - INFO - r_drop_alpha : 10.0
16
+ 2026-04-24 16:49:08,936 - INFO - infonce_temperature : 0.07
17
+ 2026-04-24 16:49:08,937 - INFO - infonce_weight : 0.5
18
+ 2026-04-24 16:49:08,937 - INFO - seed : 42
19
+ 2026-04-24 16:49:08,938 - INFO - resume_from_checkpoint : output_checkpoints/graphcodebert-base-lowLR-highBatchSize/checkpoint-500
20
+ 2026-04-24 16:49:08,939 - INFO - label_smoothing : 0.5
21
+ 2026-04-24 16:49:08,939 - INFO - adversarial_epsilon : 0.5
22
+ 2026-04-24 16:49:08,940 - INFO - use_swa : False
23
+ 2026-04-24 16:49:08,941 - INFO - swa_start_epoch : 0
24
+ 2026-04-24 16:49:08,942 - INFO - swa_lr : 1e-05
25
+ 2026-04-24 16:49:08,943 - INFO - data_augmentation : True
26
+ 2026-04-24 16:49:08,944 - INFO - aug_rename_prob : 0.8
27
+ 2026-04-24 16:49:08,944 - INFO - aug_format_prob : 0.8
28
+ 2026-04-24 16:49:08,945 - INFO - mixup_alpha : 1.0
29
+ 2026-04-24 16:49:08,946 - INFO - low_pass_keep_ratio : 0.5
30
+ 2026-04-24 16:49:08,947 - INFO - freq_consistency_weight : 0.5
31
+ 2026-04-24 16:49:08,947 - INFO - hidden_dropout_prob : 0.3
32
+ 2026-04-24 16:49:08,948 - INFO - attention_probs_dropout_prob : 0.3
33
+ 2026-04-24 16:49:08,949 - INFO - classifier_dropout : 0.3
34
+ 2026-04-24 16:49:08,949 - INFO - =================================
35
+ 2026-04-24 16:49:10,130 - INFO - Model placed on cuda
36
+ 2026-04-24 16:49:10,132 - INFO - ===== Model Architecture =====
37
+ 2026-04-24 16:49:10,134 - INFO -
38
  RobertaForSequenceClassification(
39
  (roberta): RobertaModel(
40
  (embeddings): RobertaEmbeddings(
 
79
  (out_proj): Linear(in_features=768, out_features=2, bias=True)
80
  )
81
  )
82
+ 2026-04-24 16:49:10,136 - INFO - ===== Parameter Summary =====
83
+ 2026-04-24 16:49:10,137 - INFO - Total Parameters: 124,647,170
84
+ 2026-04-24 16:49:10,138 - INFO - Trainable Parameters: 592,130
85
+ 2026-04-24 16:49:10,138 - INFO - Non-trainable Parameters: 124,055,040
86
+ 2026-04-24 16:49:10,139 - INFO - ===== Tokenizer Summary =====
87
+ 2026-04-24 16:49:10,153 - INFO - Vocab size: 50265 | Special tokens: ['<s>', '</s>', '<unk>', '<pad>', '<mask>']
88
+ 2026-04-24 16:49:10,154 - INFO - ===== End of Architecture Log =====
89
+ 2026-04-24 16:49:10,155 - INFO - Data augmentation enabled (rename=0.8, format=0.8)
90
+ 2026-04-24 16:49:44,436 - INFO - === Starting training with MixCode + FFT low-pass consistency ===