Update hyperparams.yaml
Browse files- hyperparams.yaml +2 -2
hyperparams.yaml
CHANGED
|
@@ -29,7 +29,7 @@ mean_var_norm: !new:speechbrain.processing.features.InputNormalization
|
|
| 29 |
# To design a custom model, either just edit the simple CustomModel
|
| 30 |
# class that's listed here, or replace this `!new` call with a line
|
| 31 |
# pointing to a different file you've defined.
|
| 32 |
-
embedding_model: !new:
|
| 33 |
in_channels: !ref <n_mels>
|
| 34 |
activation: !name:torch.nn.LeakyReLU
|
| 35 |
tdnn_blocks: 5
|
|
@@ -38,7 +38,7 @@ embedding_model: !new:templates.speaker_id.custom_model.Xvector
|
|
| 38 |
tdnn_dilations: [1, 2, 3, 1, 1]
|
| 39 |
lin_neurons: !ref <emb_dim>
|
| 40 |
|
| 41 |
-
classifier: !new:
|
| 42 |
input_shape: [null, null, !ref <emb_dim>]
|
| 43 |
activation: !name:torch.nn.LeakyReLU
|
| 44 |
lin_blocks: 1
|
|
|
|
| 29 |
# To design a custom model, either just edit the simple CustomModel
|
| 30 |
# class that's listed here, or replace this `!new` call with a line
|
| 31 |
# pointing to a different file you've defined.
|
| 32 |
+
embedding_model: !new:custom_model.Xvector
|
| 33 |
in_channels: !ref <n_mels>
|
| 34 |
activation: !name:torch.nn.LeakyReLU
|
| 35 |
tdnn_blocks: 5
|
|
|
|
| 38 |
tdnn_dilations: [1, 2, 3, 1, 1]
|
| 39 |
lin_neurons: !ref <emb_dim>
|
| 40 |
|
| 41 |
+
classifier: !new:custom_model.Classifier
|
| 42 |
input_shape: [null, null, !ref <emb_dim>]
|
| 43 |
activation: !name:torch.nn.LeakyReLU
|
| 44 |
lin_blocks: 1
|