2026-04-24 16:49:08,923 - INFO - ===== Training Configuration ===== 2026-04-24 16:49:08,924 - INFO - model_name : microsoft/graphcodebert-base 2026-04-24 16:49:08,924 - INFO - output_dir : output_checkpoints/graphcodebert-base-lowLR-highBatchSize/ 2026-04-24 16:49:08,925 - INFO - num_epochs : 1 2026-04-24 16:49:08,926 - INFO - max_steps : -1 2026-04-24 16:49:08,927 - INFO - batch_size : 256 2026-04-24 16:49:08,928 - INFO - learning_rate : 1e-06 2026-04-24 16:49:08,929 - INFO - max_length : 512 2026-04-24 16:49:08,930 - INFO - num_labels : 2 2026-04-24 16:49:08,931 - INFO - use_wandb : True 2026-04-24 16:49:08,931 - INFO - freeze_base : True 2026-04-24 16:49:08,932 - INFO - loss_type : r-drop 2026-04-24 16:49:08,933 - INFO - focal_alpha : 1.0 2026-04-24 16:49:08,934 - INFO - focal_gamma : 2.0 2026-04-24 16:49:08,934 - INFO - r_drop_alpha : 10.0 2026-04-24 16:49:08,936 - INFO - infonce_temperature : 0.07 2026-04-24 16:49:08,937 - INFO - infonce_weight : 0.5 2026-04-24 16:49:08,937 - INFO - seed : 42 2026-04-24 16:49:08,938 - INFO - resume_from_checkpoint : output_checkpoints/graphcodebert-base-lowLR-highBatchSize/checkpoint-500 2026-04-24 16:49:08,939 - INFO - label_smoothing : 0.5 2026-04-24 16:49:08,939 - INFO - adversarial_epsilon : 0.5 2026-04-24 16:49:08,940 - INFO - use_swa : False 2026-04-24 16:49:08,941 - INFO - swa_start_epoch : 0 2026-04-24 16:49:08,942 - INFO - swa_lr : 1e-05 2026-04-24 16:49:08,943 - INFO - data_augmentation : True 2026-04-24 16:49:08,944 - INFO - aug_rename_prob : 0.8 2026-04-24 16:49:08,944 - INFO - aug_format_prob : 0.8 2026-04-24 16:49:08,945 - INFO - mixup_alpha : 1.0 2026-04-24 16:49:08,946 - INFO - low_pass_keep_ratio : 0.5 2026-04-24 16:49:08,947 - INFO - freq_consistency_weight : 0.5 2026-04-24 16:49:08,947 - INFO - hidden_dropout_prob : 0.3 2026-04-24 16:49:08,948 - INFO - attention_probs_dropout_prob : 0.3 2026-04-24 16:49:08,949 - INFO - classifier_dropout : 0.3 2026-04-24 16:49:08,949 - INFO - ================================= 2026-04-24 16:49:10,130 - INFO - Model placed on cuda 2026-04-24 16:49:10,132 - INFO - ===== Model Architecture ===== 2026-04-24 16:49:10,134 - 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-24 16:49:10,136 - INFO - ===== Parameter Summary ===== 2026-04-24 16:49:10,137 - INFO - Total Parameters: 124,647,170 2026-04-24 16:49:10,138 - INFO - Trainable Parameters: 592,130 2026-04-24 16:49:10,138 - INFO - Non-trainable Parameters: 124,055,040 2026-04-24 16:49:10,139 - INFO - ===== Tokenizer Summary ===== 2026-04-24 16:49:10,153 - INFO - Vocab size: 50265 | Special tokens: ['', '', '', '', ''] 2026-04-24 16:49:10,154 - INFO - ===== End of Architecture Log ===== 2026-04-24 16:49:10,155 - INFO - Data augmentation enabled (rename=0.8, format=0.8) 2026-04-24 16:49:44,436 - INFO - === Starting training with MixCode + FFT low-pass consistency ===