dzungpham/graphcodebert-code-classification

12026-04-28 15:08:35,821 - INFO - ===== Training Configuration =====22026-04-28 15:08:35,822 - INFO - model_name           : microsoft/graphcodebert-base32026-04-28 15:08:35,824 - INFO - output_dir           : output_checkpoints/graphcodebert-best/42026-04-28 15:08:35,826 - INFO - num_epochs           : 252026-04-28 15:08:35,828 - INFO - max_steps            : -162026-04-28 15:08:35,829 - INFO - batch_size           : 25672026-04-28 15:08:35,831 - INFO - learning_rate        : 1e-0682026-04-28 15:08:35,832 - INFO - max_length           : 51292026-04-28 15:08:35,834 - INFO - num_labels           : 2102026-04-28 15:08:35,835 - INFO - use_wandb            : True112026-04-28 15:08:35,838 - INFO - freeze_base          : True122026-04-28 15:08:35,839 - INFO - loss_type            : r-drop132026-04-28 15:08:35,840 - INFO - focal_alpha          : 1.0142026-04-28 15:08:35,842 - INFO - focal_gamma          : 2.0152026-04-28 15:08:35,844 - INFO - r_drop_alpha         : 6.0162026-04-28 15:08:35,845 - INFO - infonce_temperature  : 0.07172026-04-28 15:08:35,846 - INFO - infonce_weight       : 0.5182026-04-28 15:08:35,849 - INFO - seed                 : 42192026-04-28 15:08:35,851 - INFO - wandb_run_name       : graphcodebert-rdrop-best202026-04-28 15:08:35,855 - INFO - resume_from_checkpoint : checkpoints/graphcodebert-best/checkpoint-1400212026-04-28 15:08:35,856 - INFO - save_steps           : 100222026-04-28 15:08:35,857 - INFO - eval_steps           : 20232026-04-28 15:08:35,858 - INFO - logging_steps        : 2242026-04-28 15:08:35,860 - INFO - label_smoothing      : 0.5252026-04-28 15:08:35,861 - INFO - adversarial_epsilon  : 0.5262026-04-28 15:08:35,862 - INFO - use_swa              : False272026-04-28 15:08:35,863 - INFO - swa_start_epoch      : 0282026-04-28 15:08:35,863 - INFO - swa_lr               : 1e-05292026-04-28 15:08:35,864 - INFO - data_augmentation    : True302026-04-28 15:08:35,865 - INFO - aug_rename_prob      : 0.8312026-04-28 15:08:35,866 - INFO - aug_format_prob      : 0.8322026-04-28 15:08:35,866 - INFO - mixup_alpha          : 1.0332026-04-28 15:08:35,868 - INFO - low_pass_keep_ratio  : 0.5342026-04-28 15:08:35,869 - INFO - freq_consistency_weight : 0.5352026-04-28 15:08:35,869 - INFO - hidden_dropout_prob  : 0.3362026-04-28 15:08:35,870 - INFO - attention_probs_dropout_prob : 0.3372026-04-28 15:08:35,871 - INFO - classifier_dropout   : 0.3382026-04-28 15:08:35,872 - INFO - =================================392026-04-28 15:08:38,268 - INFO - Model placed on cuda402026-04-28 15:08:38,273 - INFO - ===== Model Architecture =====412026-04-28 15:08:38,276 - INFO - 42RobertaForSequenceClassification(43  (roberta): RobertaModel(44    (embeddings): RobertaEmbeddings(45      (word_embeddings): Embedding(50265, 768, padding_idx=1)46      (position_embeddings): Embedding(514, 768, padding_idx=1)47      (token_type_embeddings): Embedding(1, 768)48      (LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)49      (dropout): Dropout(p=0.3, inplace=False)50    )51    (encoder): RobertaEncoder(52      (layer): ModuleList(53        (0-11): 12 x RobertaLayer(54          (attention): RobertaAttention(55            (self): RobertaSdpaSelfAttention(56              (query): Linear(in_features=768, out_features=768, bias=True)57              (key): Linear(in_features=768, out_features=768, bias=True)58              (value): Linear(in_features=768, out_features=768, bias=True)59              (dropout): Dropout(p=0.3, inplace=False)60            )61            (output): RobertaSelfOutput(62              (dense): Linear(in_features=768, out_features=768, bias=True)63              (LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)64              (dropout): Dropout(p=0.3, inplace=False)65            )66          )67          (intermediate): RobertaIntermediate(68            (dense): Linear(in_features=768, out_features=3072, bias=True)69            (intermediate_act_fn): GELUActivation()70          )71          (output): RobertaOutput(72            (dense): Linear(in_features=3072, out_features=768, bias=True)73            (LayerNorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)74            (dropout): Dropout(p=0.3, inplace=False)75          )76        )77      )78    )79  )80  (classifier): RobertaClassificationHead(81    (dense): Linear(in_features=768, out_features=768, bias=True)82    (dropout): Dropout(p=0.3, inplace=False)83    (out_proj): Linear(in_features=768, out_features=2, bias=True)84  )85)862026-04-28 15:08:38,278 - INFO - ===== Parameter Summary =====872026-04-28 15:08:38,280 - INFO - Total Parameters:         124,647,170882026-04-28 15:08:38,281 - INFO - Trainable Parameters:     592,130892026-04-28 15:08:38,281 - INFO - Non-trainable Parameters: 124,055,040902026-04-28 15:08:38,282 - INFO - ===== Tokenizer Summary =====912026-04-28 15:08:38,304 - INFO - Vocab size: 50265 | Special tokens: ['<s>', '</s>', '<unk>', '<pad>', '<mask>']922026-04-28 15:08:38,305 - INFO - ===== End of Architecture Log =====932026-04-28 15:08:38,306 - INFO - Data augmentation enabled (rename=0.8, format=0.8)942026-04-28 15:17:40,501 - INFO - === Starting training with MixCode + FFT low-pass consistency ===952026-04-28 16:02:32,342 - INFO - Training completed successfully.962026-04-28 16:02:36,702 - INFO - Final model saved to output_checkpoints/graphcodebert-best/final_model97