poonehmousavi
/

discrete_wavlm_spk_rec_ecapatdn

Model card Files Files and versions

poonehmousavi commited on Jul 30, 2024

Commit

507e23c

·

verified ·

1 Parent(s): eff61c9

Update hyperparams.yaml

Files changed (1) hide show

hyperparams.yaml +2 -2

hyperparams.yaml CHANGED Viewed

@@ -75,12 +75,12 @@ codec: !new:speechbrain.lobes.models.huggingface_transformers.discrete_ssl.Discr
     kmeans_repo_id: !ref <kmeans_repo_id>
     num_clusters: !ref <num_clusters>
-discrete_embedding_layer: !new:custom_model.Discrete_EmbeddingLayer
     num_codebooks: !ref <num_codebooks>
     vocab_size: !ref <num_clusters>
     emb_dim: !ref <encoder_dim>
-attention_mlp: !new:custom_model.AttentionMLP
     input_dim: !ref <encoder_dim>
     hidden_dim: !ref <encoder_dim>

     kmeans_repo_id: !ref <kmeans_repo_id>
     num_clusters: !ref <num_clusters>
+discrete_embedding_layer: !new:custom_interface.Discrete_EmbeddingLayer
     num_codebooks: !ref <num_codebooks>
     vocab_size: !ref <num_clusters>
     emb_dim: !ref <encoder_dim>
+attention_mlp: !new:custom_interface.AttentionMLP
     input_dim: !ref <encoder_dim>
     hidden_dim: !ref <encoder_dim>