olivierb
commited on
Commit
•
0c78d85
1
Parent(s):
9d7d11c
make hyperparams.yaml more similar to speechbrain official models
Browse files- ctc_lin.ckpt → asr.ckpt +2 -2
- enc.ckpt +0 -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:
|
3 |
-
size
|
|
|
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 |
-
|
63 |
-
|
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>
|
|