run_config.json

{
    "base_model": "docketanalyzer/docket-lm-xs",
    "run_args": {
        "max_length": 256
    },
    "training_args": {
        "num_train_epochs": 1,
        "per_device_train_batch_size": 8,
        "per_device_eval_batch_size": 16,
        "gradient_accumulation_steps": 2,
        "learning_rate": 5e-05,
        "weight_decay": 0.1,
        "warmup_steps": 30,
        "evaluation_strategy": "steps",
        "eval_steps": 60,
        "save_steps": 60,
        "save_total_limit": 2
    },
    "run_name": "initial_model",
    "run_type": "ClassificationRoutine"
}