aheba31 commited on
Commit
affacec
1 Parent(s): 3293f28

add avg_pool

Browse files
Files changed (1) hide show
  1. 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