{ "model_name": "Salesforce/codet5-base", "dataset_name": "google/code_x_glue_cc_code_refinement", "dataset_config": "small", "source_column": null, "target_column": null, "language": "java", "train_split": "train", "eval_split": "validation", "test_split": "test", "max_source_length": 256, "max_target_length": 256, "pad_to_multiple_of": 8, "learning_rate": 5e-05, "per_device_train_batch_size": 16, "per_device_eval_batch_size": 64, "gradient_accumulation_steps": 2, "num_train_epochs": 10.0, "weight_decay": 0.01, "warmup_ratio": 0.05, "lr_scheduler_type": "linear", "max_grad_norm": 1.0, "label_smoothing_factor": 0.0, "gradient_checkpointing": false, "eval_strategy": "epoch", "save_strategy": "epoch", "logging_steps": 100, "save_total_limit": 2, "load_best_model_at_end": true, "metric_for_best_model": "exact_match", "greater_is_better": true, "early_stopping_patience": 3, "eval_subset_size": 1000, "max_eval_examples": 0, "num_beams": 4, "max_new_tokens": 256, "generation_early_stopping": true, "length_penalty": 1.0, "no_repeat_ngram_size": 0, "seed": 42, "output_dir": "outputs/codet5-base-code-repair", "precision": "auto", "dataloader_num_workers": 4, "report_to": "none", "resume_from_checkpoint": null, "compute_codebleu": true, "smoke_test": false, "smoke_train_size": 256, "smoke_eval_size": 32, "smoke_epochs": 1.0 }