| { | |
| "base_model": "FacebookAI/xlm-roberta-base", | |
| "max_seq_len": 256, | |
| "use_lora": true, | |
| "lora_r": 32, | |
| "lora_alpha": 64, | |
| "lora_dropout": 0.1, | |
| "learning_rate": 0.0001, | |
| "num_train_epochs": 6, | |
| "per_device_train_batch_size": 16, | |
| "per_device_eval_batch_size": 16, | |
| "gradient_accumulation_steps": 2, | |
| "warmup_ratio": 0.15, | |
| "weight_decay": 0.01, | |
| "fp16": true, | |
| "early_stopping_patience": 5, | |
| "hp_trials": 1, | |
| "output_dir": "C:\\xampp\\htdocs\\Rise-On-AI\\scripts\\sentiment-model-training\\outputs\\best_model", | |
| "best_eval_f1_macro": 0.8157473649436856 | |
| } |