Instructions to use dzungpham/graphcodebert-code-classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dzungpham/graphcodebert-code-classification with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("dzungpham/graphcodebert-code-classification", dtype="auto") - Notebooks
- Google Colab
- Kaggle
upload unixcoder inference results
Browse files- codebert-base/checkpoint-50/config.json +28 -0
- codebert-base/checkpoint-50/config_hyperparams.json +53 -0
- codebert-base/checkpoint-50/hyperparams.json +35 -0
- codebert-base/checkpoint-50/merges.txt +0 -0
- codebert-base/checkpoint-50/model.safetensors +3 -0
- codebert-base/checkpoint-50/optimizer.pt +3 -0
- codebert-base/checkpoint-50/rng_state.pth +3 -0
- codebert-base/checkpoint-50/scaler.pt +3 -0
- codebert-base/checkpoint-50/scheduler.pt +3 -0
- codebert-base/checkpoint-50/special_tokens_map.json +51 -0
- codebert-base/checkpoint-50/tokenizer.json +0 -0
- codebert-base/checkpoint-50/tokenizer_config.json +58 -0
- codebert-base/checkpoint-50/trainer_state.json +78 -0
- codebert-base/checkpoint-50/training_args.bin +3 -0
- codebert-base/checkpoint-50/vocab.json +0 -0
- codebert-base/final_model/config.json +28 -0
- codebert-base/final_model/merges.txt +0 -0
- codebert-base/final_model/model.safetensors +3 -0
- codebert-base/final_model/special_tokens_map.json +51 -0
- codebert-base/final_model/tokenizer.json +0 -0
- codebert-base/final_model/tokenizer_config.json +58 -0
- codebert-base/final_model/vocab.json +0 -0
- codebert-base/training.log +92 -0
codebert-base/checkpoint-50/config.json
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
"hidden_act": "gelu",
|
| 11 |
+
"hidden_dropout_prob": 0.3,
|
| 12 |
+
"hidden_size": 768,
|
| 13 |
+
"initializer_range": 0.02,
|
| 14 |
+
"intermediate_size": 3072,
|
| 15 |
+
"layer_norm_eps": 1e-05,
|
| 16 |
+
"max_position_embeddings": 514,
|
| 17 |
+
"model_type": "roberta",
|
| 18 |
+
"num_attention_heads": 12,
|
| 19 |
+
"num_hidden_layers": 12,
|
| 20 |
+
"output_past": true,
|
| 21 |
+
"pad_token_id": 1,
|
| 22 |
+
"position_embedding_type": "absolute",
|
| 23 |
+
"problem_type": "single_label_classification",
|
| 24 |
+
"transformers_version": "4.56.0",
|
| 25 |
+
"type_vocab_size": 1,
|
| 26 |
+
"use_cache": true,
|
| 27 |
+
"vocab_size": 50265
|
| 28 |
+
}
|
codebert-base/checkpoint-50/config_hyperparams.json
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"train_config": {
|
| 3 |
+
"model_name": "microsoft/codebert-base",
|
| 4 |
+
"num_epochs": 0.5,
|
| 5 |
+
"batch_size": 32,
|
| 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": 6.0,
|
| 13 |
+
"infonce_temperature": 0.07,
|
| 14 |
+
"infonce_weight": 0.5,
|
| 15 |
+
"label_smoothing": 0.3,
|
| 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.6,
|
| 22 |
+
"aug_format_prob": 0.6,
|
| 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.2
|
| 29 |
+
},
|
| 30 |
+
"training_arguments": {
|
| 31 |
+
"output_dir": "output_checkpoints/codebert-base/",
|
| 32 |
+
"num_train_epochs": 0.5,
|
| 33 |
+
"per_device_train_batch_size": 32,
|
| 34 |
+
"per_device_eval_batch_size": 64,
|
| 35 |
+
"learning_rate": 1e-06,
|
| 36 |
+
"warmup_steps": 781,
|
| 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": 50,
|
| 49 |
+
"epoch": 0.006399590426212722,
|
| 50 |
+
"best_metric": null,
|
| 51 |
+
"best_model_checkpoint": null
|
| 52 |
+
}
|
| 53 |
+
}
|
codebert-base/checkpoint-50/hyperparams.json
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "microsoft/codebert-base",
|
| 3 |
+
"output_dir": "output_checkpoints/codebert-base/",
|
| 4 |
+
"num_epochs": 0.5,
|
| 5 |
+
"max_steps": 50,
|
| 6 |
+
"batch_size": 32,
|
| 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": 6.0,
|
| 16 |
+
"infonce_temperature": 0.07,
|
| 17 |
+
"infonce_weight": 0.5,
|
| 18 |
+
"seed": 42,
|
| 19 |
+
"resume_from_checkpoint": null,
|
| 20 |
+
"label_smoothing": 0.3,
|
| 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.6,
|
| 27 |
+
"aug_format_prob": 0.6,
|
| 28 |
+
"mixup_alpha": 1.0,
|
| 29 |
+
"low_pass_keep_ratio": 0.5,
|
| 30 |
+
"freq_consistency_weight": 0.2,
|
| 31 |
+
"hidden_dropout_prob": 0.3,
|
| 32 |
+
"attention_probs_dropout_prob": 0.3,
|
| 33 |
+
"classifier_dropout": 0.3,
|
| 34 |
+
"device": "cuda"
|
| 35 |
+
}
|
codebert-base/checkpoint-50/merges.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
codebert-base/checkpoint-50/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:43039dfba735b6122116023d874be54ad6afdcea0c49ac8cd3ddf7ca7f697196
|
| 3 |
+
size 498612824
|
codebert-base/checkpoint-50/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1b4ad0b0c7a0a0b11cb40991d29ab1620733f8bcc4ca8ad7243c39080e950d0a
|
| 3 |
+
size 4741859
|
codebert-base/checkpoint-50/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1a9ab4f75b99203755841ed328e95e73d1e090a77e9b99ba1e6303dd9271f635
|
| 3 |
+
size 14645
|
codebert-base/checkpoint-50/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b21c5349d5e7d02de630ebc1cb53ade1d9c6079eeb8594d223bb786011a0428b
|
| 3 |
+
size 1383
|
codebert-base/checkpoint-50/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ea2915d03c29503502184966928010b43e225be9484b2fb51440baa31032967c
|
| 3 |
+
size 1465
|
codebert-base/checkpoint-50/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": false,
|
| 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 |
+
}
|
codebert-base/checkpoint-50/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
codebert-base/checkpoint-50/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": false,
|
| 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 |
+
}
|
codebert-base/checkpoint-50/trainer_state.json
ADDED
|
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 0.006399590426212722,
|
| 6 |
+
"eval_steps": 1000,
|
| 7 |
+
"global_step": 50,
|
| 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.0012799180852425445,
|
| 14 |
+
"grad_norm": 295869.8125,
|
| 15 |
+
"learning_rate": 1.1523687580025607e-08,
|
| 16 |
+
"loss": 0.8874,
|
| 17 |
+
"step": 10
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.002559836170485089,
|
| 21 |
+
"grad_norm": 442576.28125,
|
| 22 |
+
"learning_rate": 2.4327784891165172e-08,
|
| 23 |
+
"loss": 0.8949,
|
| 24 |
+
"step": 20
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.0038397542557276334,
|
| 28 |
+
"grad_norm": 390849.34375,
|
| 29 |
+
"learning_rate": 3.713188220230474e-08,
|
| 30 |
+
"loss": 0.898,
|
| 31 |
+
"step": 30
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.005119672340970178,
|
| 35 |
+
"grad_norm": 279607.9375,
|
| 36 |
+
"learning_rate": 4.9935979513444295e-08,
|
| 37 |
+
"loss": 0.8886,
|
| 38 |
+
"step": 40
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.006399590426212722,
|
| 42 |
+
"grad_norm": 344149.03125,
|
| 43 |
+
"learning_rate": 6.274007682458386e-08,
|
| 44 |
+
"loss": 0.8973,
|
| 45 |
+
"step": 50
|
| 46 |
+
}
|
| 47 |
+
],
|
| 48 |
+
"logging_steps": 10,
|
| 49 |
+
"max_steps": 50,
|
| 50 |
+
"num_input_tokens_seen": 0,
|
| 51 |
+
"num_train_epochs": 1,
|
| 52 |
+
"save_steps": 50,
|
| 53 |
+
"stateful_callbacks": {
|
| 54 |
+
"EarlyStoppingCallback": {
|
| 55 |
+
"args": {
|
| 56 |
+
"early_stopping_patience": 3,
|
| 57 |
+
"early_stopping_threshold": 0.0
|
| 58 |
+
},
|
| 59 |
+
"attributes": {
|
| 60 |
+
"early_stopping_patience_counter": 0
|
| 61 |
+
}
|
| 62 |
+
},
|
| 63 |
+
"TrainerControl": {
|
| 64 |
+
"args": {
|
| 65 |
+
"should_epoch_stop": false,
|
| 66 |
+
"should_evaluate": false,
|
| 67 |
+
"should_log": false,
|
| 68 |
+
"should_save": true,
|
| 69 |
+
"should_training_stop": true
|
| 70 |
+
},
|
| 71 |
+
"attributes": {}
|
| 72 |
+
}
|
| 73 |
+
},
|
| 74 |
+
"total_flos": 841955377152000.0,
|
| 75 |
+
"train_batch_size": 64,
|
| 76 |
+
"trial_name": null,
|
| 77 |
+
"trial_params": null
|
| 78 |
+
}
|
codebert-base/checkpoint-50/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:716747561b45f935ce7bd12c1883ff0618d3af217eb0ce2bf14bba6f805506ea
|
| 3 |
+
size 5841
|
codebert-base/checkpoint-50/vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
codebert-base/final_model/config.json
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
"hidden_act": "gelu",
|
| 11 |
+
"hidden_dropout_prob": 0.3,
|
| 12 |
+
"hidden_size": 768,
|
| 13 |
+
"initializer_range": 0.02,
|
| 14 |
+
"intermediate_size": 3072,
|
| 15 |
+
"layer_norm_eps": 1e-05,
|
| 16 |
+
"max_position_embeddings": 514,
|
| 17 |
+
"model_type": "roberta",
|
| 18 |
+
"num_attention_heads": 12,
|
| 19 |
+
"num_hidden_layers": 12,
|
| 20 |
+
"output_past": true,
|
| 21 |
+
"pad_token_id": 1,
|
| 22 |
+
"position_embedding_type": "absolute",
|
| 23 |
+
"problem_type": "single_label_classification",
|
| 24 |
+
"transformers_version": "4.56.0",
|
| 25 |
+
"type_vocab_size": 1,
|
| 26 |
+
"use_cache": true,
|
| 27 |
+
"vocab_size": 50265
|
| 28 |
+
}
|
codebert-base/final_model/merges.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
codebert-base/final_model/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:43039dfba735b6122116023d874be54ad6afdcea0c49ac8cd3ddf7ca7f697196
|
| 3 |
+
size 498612824
|
codebert-base/final_model/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": false,
|
| 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 |
+
}
|
codebert-base/final_model/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
codebert-base/final_model/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": false,
|
| 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 |
+
}
|
codebert-base/final_model/vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
codebert-base/training.log
ADDED
|
@@ -0,0 +1,92 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-04-23 14:58:06,425 - INFO - ===== Training Configuration =====
|
| 2 |
+
2026-04-23 14:58:06,427 - INFO - model_name : microsoft/codebert-base
|
| 3 |
+
2026-04-23 14:58:06,428 - INFO - output_dir : output_checkpoints/codebert-base/
|
| 4 |
+
2026-04-23 14:58:06,429 - INFO - num_epochs : 0.5
|
| 5 |
+
2026-04-23 14:58:06,431 - INFO - max_steps : 50
|
| 6 |
+
2026-04-23 14:58:06,432 - INFO - batch_size : 32
|
| 7 |
+
2026-04-23 14:58:06,433 - INFO - learning_rate : 1e-06
|
| 8 |
+
2026-04-23 14:58:06,434 - INFO - max_length : 512
|
| 9 |
+
2026-04-23 14:58:06,436 - INFO - num_labels : 2
|
| 10 |
+
2026-04-23 14:58:06,437 - INFO - use_wandb : True
|
| 11 |
+
2026-04-23 14:58:06,439 - INFO - freeze_base : True
|
| 12 |
+
2026-04-23 14:58:06,440 - INFO - loss_type : r-drop
|
| 13 |
+
2026-04-23 14:58:06,442 - INFO - focal_alpha : 1.0
|
| 14 |
+
2026-04-23 14:58:06,443 - INFO - focal_gamma : 2.0
|
| 15 |
+
2026-04-23 14:58:06,444 - INFO - r_drop_alpha : 6.0
|
| 16 |
+
2026-04-23 14:58:06,446 - INFO - infonce_temperature : 0.07
|
| 17 |
+
2026-04-23 14:58:06,447 - INFO - infonce_weight : 0.5
|
| 18 |
+
2026-04-23 14:58:06,448 - INFO - seed : 42
|
| 19 |
+
2026-04-23 14:58:06,449 - INFO - resume_from_checkpoint : None
|
| 20 |
+
2026-04-23 14:58:06,451 - INFO - label_smoothing : 0.3
|
| 21 |
+
2026-04-23 14:58:06,452 - INFO - adversarial_epsilon : 0.5
|
| 22 |
+
2026-04-23 14:58:06,453 - INFO - use_swa : False
|
| 23 |
+
2026-04-23 14:58:06,454 - INFO - swa_start_epoch : 0
|
| 24 |
+
2026-04-23 14:58:06,455 - INFO - swa_lr : 1e-05
|
| 25 |
+
2026-04-23 14:58:06,457 - INFO - data_augmentation : True
|
| 26 |
+
2026-04-23 14:58:06,458 - INFO - aug_rename_prob : 0.6
|
| 27 |
+
2026-04-23 14:58:06,459 - INFO - aug_format_prob : 0.6
|
| 28 |
+
2026-04-23 14:58:06,460 - INFO - mixup_alpha : 1.0
|
| 29 |
+
2026-04-23 14:58:06,462 - INFO - low_pass_keep_ratio : 0.5
|
| 30 |
+
2026-04-23 14:58:06,463 - INFO - freq_consistency_weight : 0.2
|
| 31 |
+
2026-04-23 14:58:06,464 - INFO - hidden_dropout_prob : 0.3
|
| 32 |
+
2026-04-23 14:58:06,466 - INFO - attention_probs_dropout_prob : 0.3
|
| 33 |
+
2026-04-23 14:58:06,467 - INFO - classifier_dropout : 0.3
|
| 34 |
+
2026-04-23 14:58:06,469 - INFO - =================================
|
| 35 |
+
2026-04-23 14:58:13,859 - INFO - Model placed on cuda
|
| 36 |
+
2026-04-23 14:58:13,865 - INFO - ===== Model Architecture =====
|
| 37 |
+
2026-04-23 14:58:13,867 - INFO -
|
| 38 |
+
RobertaForSequenceClassification(
|
| 39 |
+
(roberta): RobertaModel(
|
| 40 |
+
(embeddings): RobertaEmbeddings(
|
| 41 |
+
(word_embeddings): Embedding(50265, 768, padding_idx=1)
|
| 42 |
+
(position_embeddings): Embedding(514, 768, padding_idx=1)
|
| 43 |
+
(token_type_embeddings): Embedding(1, 768)
|
| 44 |
+
(LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)
|
| 45 |
+
(dropout): Dropout(p=0.3, inplace=False)
|
| 46 |
+
)
|
| 47 |
+
(encoder): RobertaEncoder(
|
| 48 |
+
(layer): ModuleList(
|
| 49 |
+
(0-11): 12 x RobertaLayer(
|
| 50 |
+
(attention): RobertaAttention(
|
| 51 |
+
(self): RobertaSdpaSelfAttention(
|
| 52 |
+
(query): Linear(in_features=768, out_features=768, bias=True)
|
| 53 |
+
(key): Linear(in_features=768, out_features=768, bias=True)
|
| 54 |
+
(value): Linear(in_features=768, out_features=768, bias=True)
|
| 55 |
+
(dropout): Dropout(p=0.3, inplace=False)
|
| 56 |
+
)
|
| 57 |
+
(output): RobertaSelfOutput(
|
| 58 |
+
(dense): Linear(in_features=768, out_features=768, bias=True)
|
| 59 |
+
(LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)
|
| 60 |
+
(dropout): Dropout(p=0.3, inplace=False)
|
| 61 |
+
)
|
| 62 |
+
)
|
| 63 |
+
(intermediate): RobertaIntermediate(
|
| 64 |
+
(dense): Linear(in_features=768, out_features=3072, bias=True)
|
| 65 |
+
(intermediate_act_fn): GELUActivation()
|
| 66 |
+
)
|
| 67 |
+
(output): RobertaOutput(
|
| 68 |
+
(dense): Linear(in_features=3072, out_features=768, bias=True)
|
| 69 |
+
(LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)
|
| 70 |
+
(dropout): Dropout(p=0.3, inplace=False)
|
| 71 |
+
)
|
| 72 |
+
)
|
| 73 |
+
)
|
| 74 |
+
)
|
| 75 |
+
)
|
| 76 |
+
(classifier): RobertaClassificationHead(
|
| 77 |
+
(dense): Linear(in_features=768, out_features=768, bias=True)
|
| 78 |
+
(dropout): Dropout(p=0.3, inplace=False)
|
| 79 |
+
(out_proj): Linear(in_features=768, out_features=2, bias=True)
|
| 80 |
+
)
|
| 81 |
+
)
|
| 82 |
+
2026-04-23 14:58:13,869 - INFO - ===== Parameter Summary =====
|
| 83 |
+
2026-04-23 14:58:13,871 - INFO - Total Parameters: 124,647,170
|
| 84 |
+
2026-04-23 14:58:13,872 - INFO - Trainable Parameters: 592,130
|
| 85 |
+
2026-04-23 14:58:13,874 - INFO - Non-trainable Parameters: 124,055,040
|
| 86 |
+
2026-04-23 14:58:13,875 - INFO - ===== Tokenizer Summary =====
|
| 87 |
+
2026-04-23 14:58:13,888 - INFO - Vocab size: 50265 | Special tokens: ['<s>', '</s>', '<unk>', '<pad>', '<mask>']
|
| 88 |
+
2026-04-23 14:58:13,889 - INFO - ===== End of Architecture Log =====
|
| 89 |
+
2026-04-23 14:58:13,890 - INFO - Data augmentation enabled (rename=0.6, format=0.6)
|
| 90 |
+
2026-04-23 15:01:32,389 - INFO - === Starting training with MixCode + FFT low-pass consistency ===
|
| 91 |
+
2026-04-23 15:03:22,966 - INFO - Training complete!
|
| 92 |
+
2026-04-23 15:03:23,707 - INFO - Final model saved to output_checkpoints/codebert-base/final_model
|