| { | |
| "model_name": "bert-base-uncased", | |
| "text_columns": ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11"], | |
| "target_column": "command", | |
| "num_train_epochs": 3, | |
| "per_device_train_batch_size": 8, | |
| "per_device_eval_batch_size": 8, | |
| "warmup_ratio": 0.1, | |
| "weight_decay": 0.01, | |
| "learning_rate": 5e-5, | |
| "fp16": false, | |
| "push_to_hub": true, | |
| "hub_model_id": "your_model_name_on_hub" | |
| } |