config.json

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