| { | |
| "whisper_model_name": "openai/whisper-small", | |
| "pooling_strategy": "learned_pooling", | |
| "sequence_len": 1500, | |
| "validation_split": 0.15, | |
| "num_epochs": 80, | |
| "learning_rate": 5e-4, | |
| "min_learning_rate": 1e-6, | |
| "dropout": 0.2, | |
| "classifier_depth": 4, | |
| "chunk_size_max_min_seconds": [1, 10], | |
| "use_mixed_precision": true, | |
| "input_dims": 768, | |
| "num_classes": 2, | |
| "warmup_epochs": 5, | |
| "training": { | |
| "wandb_project": "voicemail-classification", | |
| "wandb_run_name": "whisper-small-classifier", | |
| "save_model_path": "classifier.pth", | |
| "data_dir": "./temp" | |
| }, | |
| "whisper_config": { | |
| "sampling_rate": 16000, | |
| "max_source_positions": 1500 | |
| } | |
| } |