Adel-Moumen commited on
Commit
28923c2
1 Parent(s): 6fbf399

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +3 -1
hyperparams.yaml CHANGED
@@ -51,7 +51,7 @@ enc: !new:speechbrain.nnet.containers.Sequential
51
  bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
52
  activation3: !new:torch.nn.LeakyReLU
53
 
54
- wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
55
  source: !ref <wav2vec2_hub>
56
  output_norm: True
57
  freeze: True
@@ -76,6 +76,8 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
76
  wav2vec2: !ref <wav2vec2>
77
  enc: !ref <enc>
78
  ctc_lin: !ref <ctc_lin>
 
 
79
 
80
  modules:
81
  encoder: !ref <encoder>
 
51
  bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
52
  activation3: !new:torch.nn.LeakyReLU
53
 
54
+ wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
55
  source: !ref <wav2vec2_hub>
56
  output_norm: True
57
  freeze: True
 
76
  wav2vec2: !ref <wav2vec2>
77
  enc: !ref <enc>
78
  ctc_lin: !ref <ctc_lin>
79
+ log_softmax: !ref <log_softmax>
80
+
81
 
82
  modules:
83
  encoder: !ref <encoder>