PROPS_K16 / config.json
ThomasThebaud's picture
Add PROPS K16 checkpoint, model card, and inference code
a4534ab verified
Raw History Blame Contribute Delete
591 Bytes
{
"architectures": ["ComposedGMMMDN"],
"model_type": "props-composed-gmm-mdn",
"input_dim": 384,
"output_dim": 192,
"requested_components_per_profile": 16,
"effective_components": 15072,
"hidden_dims": [1024, 2048, 1024],
"dropout": 0.1,
"min_sigma": 0.0001,
"text_encoder": "sentence-transformers/all-MiniLM-L6-v2",
"speaker_embedding_model": "speechbrain/spkrec-ecapa-voxceleb",
"checkpoint_file": "16_components_Capspeech_min100_p=1_finetune.pt",
"checkpoint_epoch": 5,
"checkpoint_dev_nll": -310.3393858413023,
"checkpoint_train_loss": -343.3753071932471
}