dzungpham/graphcodebert-code-classification
0
12026-04-24 17:43:47,257 - INFO - ===== Training Configuration =====22026-04-24 17:43:47,259 - INFO - model_name : microsoft/graphcodebert-base32026-04-24 17:43:47,260 - INFO - output_dir : output_checkpoints/graphcodebert-base-lowLR-highBatchSize/42026-04-24 17:43:47,261 - INFO - num_epochs : 152026-04-24 17:43:47,261 - INFO - max_steps : -162026-04-24 17:43:47,262 - INFO - batch_size : 25672026-04-24 17:43:47,264 - INFO - learning_rate : 1e-0682026-04-24 17:43:47,265 - INFO - max_length : 51292026-04-24 17:43:47,266 - INFO - num_labels : 2102026-04-24 17:43:47,267 - INFO - use_wandb : True112026-04-24 17:43:47,268 - INFO - freeze_base : True122026-04-24 17:43:47,269 - INFO - loss_type : r-drop132026-04-24 17:43:47,270 - INFO - focal_alpha : 1.0142026-04-24 17:43:47,271 - INFO - focal_gamma : 2.0152026-04-24 17:43:47,272 - INFO - r_drop_alpha : 10.0162026-04-24 17:43:47,273 - INFO - infonce_temperature : 0.07172026-04-24 17:43:47,274 - INFO - infonce_weight : 0.5182026-04-24 17:43:47,275 - INFO - seed : 42192026-04-24 17:43:47,276 - INFO - resume_from_checkpoint : checkpoints/graphcodebert-base-lowLR-highBatchSize/checkpoint-500202026-04-24 17:43:47,276 - INFO - label_smoothing : 0.5212026-04-24 17:43:47,277 - INFO - adversarial_epsilon : 0.5222026-04-24 17:43:47,279 - INFO - use_swa : False232026-04-24 17:43:47,279 - INFO - swa_start_epoch : 0242026-04-24 17:43:47,280 - INFO - swa_lr : 1e-05252026-04-24 17:43:47,281 - INFO - data_augmentation : True262026-04-24 17:43:47,282 - INFO - aug_rename_prob : 0.8272026-04-24 17:43:47,283 - INFO - aug_format_prob : 0.8282026-04-24 17:43:47,285 - INFO - mixup_alpha : 1.0292026-04-24 17:43:47,285 - INFO - low_pass_keep_ratio : 0.5302026-04-24 17:43:47,286 - INFO - freq_consistency_weight : 0.5312026-04-24 17:43:47,288 - INFO - hidden_dropout_prob : 0.3322026-04-24 17:43:47,289 - INFO - attention_probs_dropout_prob : 0.3332026-04-24 17:43:47,290 - INFO - classifier_dropout : 0.3342026-04-24 17:43:47,290 - INFO - =================================352026-04-24 17:43:53,414 - INFO - Model placed on cuda362026-04-24 17:43:53,416 - INFO - ===== Model Architecture =====372026-04-24 17:43:53,419 - INFO - 38RobertaForSequenceClassification(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)822026-04-24 17:43:53,420 - INFO - ===== Parameter Summary =====832026-04-24 17:43:53,421 - INFO - Total Parameters: 124,647,170842026-04-24 17:43:53,422 - INFO - Trainable Parameters: 592,130852026-04-24 17:43:53,423 - INFO - Non-trainable Parameters: 124,055,040862026-04-24 17:43:53,423 - INFO - ===== Tokenizer Summary =====872026-04-24 17:43:53,438 - INFO - Vocab size: 50265 | Special tokens: ['<s>', '</s>', '<unk>', '<pad>', '<mask>']882026-04-24 17:43:53,438 - INFO - ===== End of Architecture Log =====892026-04-24 17:43:53,439 - INFO - Data augmentation enabled (rename=0.8, format=0.8)902026-04-24 17:47:51,059 - INFO - === Starting training with MixCode + FFT low-pass consistency ===912026-04-24 20:25:51,228 - INFO - Training completed successfully.922026-04-24 20:25:52,599 - INFO - Final model saved to output_checkpoints/graphcodebert-base-lowLR-highBatchSize/final_model93 