| { | |
| "model_type": "light", | |
| "input_dim": 9611, | |
| "output_dim": 64, | |
| "batch_size": 32, | |
| "epochs": 200, | |
| "learning_rate": 0.001, | |
| "weight_decay": 0.0001, | |
| "loss_type": "mse", | |
| "loss_alpha": 0.8, | |
| "scheduler_type": "cosine", | |
| "patience": 30, | |
| "device": "cuda", | |
| "use_amp": true, | |
| "seed": 42 | |
| } |