Adel-Moumen commited on
Commit
02e1f30
1 Parent(s): fbcc994

Update hyperparams.yaml

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