VKR_Model / training_args.json
Kirill
Add VKR CodeBERT detector model
aa7f273
{
"train": "data\\splits\\train.jsonl",
"val": "data\\splits\\val.jsonl",
"output_dir": "runs\\codebert-ai-detector",
"model_name": "microsoft/codebert-base",
"max_length": 512,
"epochs": 3,
"batch_size": 8,
"gradient_accumulation_steps": 1,
"learning_rate": 2e-05,
"weight_decay": 0.01,
"warmup_ratio": 0.06,
"seed": 42,
"freeze_encoder": true,
"limit_train": null,
"limit_val": null,
"num_workers": 0
}