olivierb commited on
Commit
0c78d85
1 Parent(s): 9d7d11c

make hyperparams.yaml more similar to speechbrain official models

Browse files
Files changed (3) hide show
  1. ctc_lin.ckpt → asr.ckpt +2 -2
  2. enc.ckpt +0 -3
  3. hyperparams.yaml +5 -3
ctc_lin.ckpt → asr.ckpt RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0ab1796261dddd41f3d25a7444d6e229475b0a46c935917578363aa440bcc6d7
3
- size 2050983
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb2d00a6a29557628ab599a7752d315ef0a56cfb2b2a8c7fb87bc4bec4c35ac3
3
+ size 14700645
enc.ckpt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8badca04f7878950012cfbed3864b263e3326d34232cc171a2bc466e78b91490
3
- size 12650406
 
 
 
 
hyperparams.yaml CHANGED
@@ -56,9 +56,11 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
56
  modules:
57
  encoder: !ref <encoder>
58
 
 
 
 
59
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
60
  loadables:
61
  wav2vec2: !ref <wav2vec2>
62
- enc: !ref <enc>
63
- ctc_lin: !ref <ctc_lin>
64
- tokenizer: !ref <tokenizer>
 
56
  modules:
57
  encoder: !ref <encoder>
58
 
59
+ asr_model: !new:torch.nn.ModuleList
60
+ - [!ref <enc>, !ref <ctc_lin>]
61
+
62
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
63
  loadables:
64
  wav2vec2: !ref <wav2vec2>
65
+ asr: !ref <asr_model>
66
+ tokenizer: !ref <tokenizer>