| { | |
| "model_type": "wav2vec2_classifier", | |
| "base_model": "facebook/wav2vec2-base", | |
| "num_classes": 2, | |
| "class_names": ["background", "switch"], | |
| "sample_rate": 16000, | |
| "max_length": 80000, | |
| "feature_size": 768, | |
| "classifier_config": { | |
| "hidden_size": 256, | |
| "dropout": 0.3, | |
| "num_layers": 2 | |
| }, | |
| "training_config": { | |
| "epochs": 15, | |
| "learning_rate": 1e-4, | |
| "batch_size": 4, | |
| "optimizer": "AdamW", | |
| "loss_function": "CrossEntropyLoss" | |
| }, | |
| "performance": { | |
| "accuracy": 1.0, | |
| "precision": 1.0, | |
| "recall": 1.0, | |
| "f1_score": 1.0 | |
| } | |
| } |