Adel-Moumen commited on
Commit
1249bc1
1 Parent(s): 177a941

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +2 -1
hyperparams.yaml CHANGED
@@ -48,7 +48,7 @@ enc: !new:speechbrain.nnet.containers.Sequential
48
  bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
49
  activation3: !new:torch.nn.LeakyReLU
50
 
51
- wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
52
  source: !ref <wav2vec2_hub>
53
  output_norm: True
54
  freeze: !ref <freeze_wav2vec>
@@ -73,6 +73,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
73
  wav2vec2: !ref <wav2vec2>
74
  enc: !ref <enc>
75
  ctc_lin: !ref <ctc_lin>
 
76
 
77
  modules:
78
  encoder: !ref <encoder>
48
  bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
49
  activation3: !new:torch.nn.LeakyReLU
50
 
51
+ wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
52
  source: !ref <wav2vec2_hub>
53
  output_norm: True
54
  freeze: !ref <freeze_wav2vec>
73
  wav2vec2: !ref <wav2vec2>
74
  enc: !ref <enc>
75
  ctc_lin: !ref <ctc_lin>
76
+ log_softmax: !ref <log_softmax>
77
 
78
  modules:
79
  encoder: !ref <encoder>