| { | |
| "base_model": "docketanalyzer/docket-lm-xs", | |
| "run_args": { | |
| "max_length": 256 | |
| }, | |
| "training_args": { | |
| "num_train_epochs": 6, | |
| "per_device_train_batch_size": 16, | |
| "per_device_eval_batch_size": 16, | |
| "gradient_accumulation_steps": 1, | |
| "learning_rate": 5e-05, | |
| "weight_decay": 0.1, | |
| "warmup_ratio": 0.02, | |
| "evaluation_strategy": "steps", | |
| "eval_steps": 60, | |
| "save_steps": 60, | |
| "save_total_limit": 1 | |
| }, | |
| "run_name": "complaint", | |
| "run_type": "ClassificationRoutine" | |
| } |