aheba31 commited on
Commit
3bfc49f
1 Parent(s): affacec

add softmax

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +2 -0
hyperparams.yaml CHANGED
@@ -36,6 +36,8 @@ modules:
36
  output_mlp: !ref <output_mlp>
37
  avg_pool: !ref <avg_pool>
38
 
 
 
39
 
40
  label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
41
 
 
36
  output_mlp: !ref <output_mlp>
37
  avg_pool: !ref <avg_pool>
38
 
39
+ softmax: !new:speechbrain.nnet.activations.Softmax
40
+
41
 
42
  label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
43