| 2026-04-28 15:08:35,821 - INFO - ===== Training Configuration ===== |
| 2026-04-28 15:08:35,822 - INFO - model_name : microsoft/graphcodebert-base |
| 2026-04-28 15:08:35,824 - INFO - output_dir : output_checkpoints/graphcodebert-best/ |
| 2026-04-28 15:08:35,826 - INFO - num_epochs : 2 |
| 2026-04-28 15:08:35,828 - INFO - max_steps : -1 |
| 2026-04-28 15:08:35,829 - INFO - batch_size : 256 |
| 2026-04-28 15:08:35,831 - INFO - learning_rate : 1e-06 |
| 2026-04-28 15:08:35,832 - INFO - max_length : 512 |
| 2026-04-28 15:08:35,834 - INFO - num_labels : 2 |
| 2026-04-28 15:08:35,835 - INFO - use_wandb : True |
| 2026-04-28 15:08:35,838 - INFO - freeze_base : True |
| 2026-04-28 15:08:35,839 - INFO - loss_type : r-drop |
| 2026-04-28 15:08:35,840 - INFO - focal_alpha : 1.0 |
| 2026-04-28 15:08:35,842 - INFO - focal_gamma : 2.0 |
| 2026-04-28 15:08:35,844 - INFO - r_drop_alpha : 6.0 |
| 2026-04-28 15:08:35,845 - INFO - infonce_temperature : 0.07 |
| 2026-04-28 15:08:35,846 - INFO - infonce_weight : 0.5 |
| 2026-04-28 15:08:35,849 - INFO - seed : 42 |
| 2026-04-28 15:08:35,851 - INFO - wandb_run_name : graphcodebert-rdrop-best |
| 2026-04-28 15:08:35,855 - INFO - resume_from_checkpoint : checkpoints/graphcodebert-best/checkpoint-1400 |
| 2026-04-28 15:08:35,856 - INFO - save_steps : 100 |
| 2026-04-28 15:08:35,857 - INFO - eval_steps : 20 |
| 2026-04-28 15:08:35,858 - INFO - logging_steps : 2 |
| 2026-04-28 15:08:35,860 - INFO - label_smoothing : 0.5 |
| 2026-04-28 15:08:35,861 - INFO - adversarial_epsilon : 0.5 |
| 2026-04-28 15:08:35,862 - INFO - use_swa : False |
| 2026-04-28 15:08:35,863 - INFO - swa_start_epoch : 0 |
| 2026-04-28 15:08:35,863 - INFO - swa_lr : 1e-05 |
| 2026-04-28 15:08:35,864 - INFO - data_augmentation : True |
| 2026-04-28 15:08:35,865 - INFO - aug_rename_prob : 0.8 |
| 2026-04-28 15:08:35,866 - INFO - aug_format_prob : 0.8 |
| 2026-04-28 15:08:35,866 - INFO - mixup_alpha : 1.0 |
| 2026-04-28 15:08:35,868 - INFO - low_pass_keep_ratio : 0.5 |
| 2026-04-28 15:08:35,869 - INFO - freq_consistency_weight : 0.5 |
| 2026-04-28 15:08:35,869 - INFO - hidden_dropout_prob : 0.3 |
| 2026-04-28 15:08:35,870 - INFO - attention_probs_dropout_prob : 0.3 |
| 2026-04-28 15:08:35,871 - INFO - classifier_dropout : 0.3 |
| 2026-04-28 15:08:35,872 - INFO - ================================= |
| 2026-04-28 15:08:38,268 - INFO - Model placed on cuda |
| 2026-04-28 15:08:38,273 - INFO - ===== Model Architecture ===== |
| 2026-04-28 15:08:38,276 - INFO - |
| RobertaForSequenceClassification( |
| (roberta): RobertaModel( |
| (embeddings): RobertaEmbeddings( |
| (word_embeddings): Embedding(50265, 768, padding_idx=1) |
| (position_embeddings): Embedding(514, 768, padding_idx=1) |
| (token_type_embeddings): Embedding(1, 768) |
| (LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True) |
| (dropout): Dropout(p=0.3, inplace=False) |
| ) |
| (encoder): RobertaEncoder( |
| (layer): ModuleList( |
| (0-11): 12 x RobertaLayer( |
| (attention): RobertaAttention( |
| (self): RobertaSdpaSelfAttention( |
| (query): Linear(in_features=768, out_features=768, bias=True) |
| (key): Linear(in_features=768, out_features=768, bias=True) |
| (value): Linear(in_features=768, out_features=768, bias=True) |
| (dropout): Dropout(p=0.3, inplace=False) |
| ) |
| (output): RobertaSelfOutput( |
| (dense): Linear(in_features=768, out_features=768, bias=True) |
| (LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True) |
| (dropout): Dropout(p=0.3, inplace=False) |
| ) |
| ) |
| (intermediate): RobertaIntermediate( |
| (dense): Linear(in_features=768, out_features=3072, bias=True) |
| (intermediate_act_fn): GELUActivation() |
| ) |
| (output): RobertaOutput( |
| (dense): Linear(in_features=3072, out_features=768, bias=True) |
| (LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True) |
| (dropout): Dropout(p=0.3, inplace=False) |
| ) |
| ) |
| ) |
| ) |
| ) |
| (classifier): RobertaClassificationHead( |
| (dense): Linear(in_features=768, out_features=768, bias=True) |
| (dropout): Dropout(p=0.3, inplace=False) |
| (out_proj): Linear(in_features=768, out_features=2, bias=True) |
| ) |
| ) |
| 2026-04-28 15:08:38,278 - INFO - ===== Parameter Summary ===== |
| 2026-04-28 15:08:38,280 - INFO - Total Parameters: 124,647,170 |
| 2026-04-28 15:08:38,281 - INFO - Trainable Parameters: 592,130 |
| 2026-04-28 15:08:38,281 - INFO - Non-trainable Parameters: 124,055,040 |
| 2026-04-28 15:08:38,282 - INFO - ===== Tokenizer Summary ===== |
| 2026-04-28 15:08:38,304 - INFO - Vocab size: 50265 | Special tokens: ['<s>', '</s>', '<unk>', '<pad>', '<mask>'] |
| 2026-04-28 15:08:38,305 - INFO - ===== End of Architecture Log ===== |
| 2026-04-28 15:08:38,306 - INFO - Data augmentation enabled (rename=0.8, format=0.8) |
| 2026-04-28 15:17:40,501 - INFO - === Starting training with MixCode + FFT low-pass consistency === |
| 2026-04-28 16:02:32,342 - INFO - Training completed successfully. |
| 2026-04-28 16:02:36,702 - INFO - Final model saved to output_checkpoints/graphcodebert-best/final_model |
|
|
