config.json

{
    "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
    }
}