add avg_pool
Browse files- hyperparams.yaml +1 -0
hyperparams.yaml
CHANGED
|
@@ -34,6 +34,7 @@ model: !new:torch.nn.ModuleList
|
|
| 34 |
modules:
|
| 35 |
wav2vec2: !ref <wav2vec2>
|
| 36 |
output_mlp: !ref <output_mlp>
|
|
|
|
| 37 |
|
| 38 |
|
| 39 |
label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
|
|
|
|
| 34 |
modules:
|
| 35 |
wav2vec2: !ref <wav2vec2>
|
| 36 |
output_mlp: !ref <output_mlp>
|
| 37 |
+
avg_pool: !ref <avg_pool>
|
| 38 |
|
| 39 |
|
| 40 |
label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
|