mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-04 14:44:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change configuration from M to L

Browse Source
This commit is contained in:
Daniel Povey 2022-05-12 16:10:08 +08:00
parent 0bf538a4a3
commit c4c9b8cf80
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned2_knowledge/conformer.py
View File

@ -63,7 +63,7 @@ class Conformer(EncoderInterface):
cnn_module_kernel: int = 31,
knowledge_M: int = 256,
knowledge_N: int = 2,
knowledge_D: int = 256,
knowledge_D: int = 512,
knowledge_K: int = 16,
) -> None:
super(Conformer, self).__init__()
@ -178,7 +178,7 @@ class ConformerEncoderLayer(nn.Module):
cnn_module_kernel: int = 31,
knowledge_M: int = 256,
knowledge_N: int = 2,
knowledge_D: int = 256,
knowledge_D: int = 512,
knowledge_K: int = 16,
) -> None:
super(ConformerEncoderLayer, self).__init__()

4
egs/librispeech/ASR/pruned2_knowledge/train.py
View File

@ -322,9 +322,9 @@ def get_params() -> AttributeDict:
# parameters for conformer
"feature_dim": 80,
"subsampling_factor": 4,
"encoder_dim": 256,
"encoder_dim": 512,
"nhead": 4,
"dim_feedforward": 1024,
"dim_feedforward": 2048,
"num_encoder_layers": 18,
# parameters for decoder
"decoder_dim": 512,