mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-07 08:04: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, cnn_module_kernel: int = 31,
knowledge_M: int = 256, knowledge_M: int = 256,
knowledge_N: int = 2, knowledge_N: int = 2,
knowledge_D: int = 256, knowledge_D: int = 512,
knowledge_K: int = 16, knowledge_K: int = 16,
) -> None: ) -> None:
super(Conformer, self).__init__() super(Conformer, self).__init__()
@ -178,7 +178,7 @@ class ConformerEncoderLayer(nn.Module):
cnn_module_kernel: int = 31, cnn_module_kernel: int = 31,
knowledge_M: int = 256, knowledge_M: int = 256,
knowledge_N: int = 2, knowledge_N: int = 2,
knowledge_D: int = 256, knowledge_D: int = 512,
knowledge_K: int = 16, knowledge_K: int = 16,
) -> None: ) -> None:
super(ConformerEncoderLayer, self).__init__() 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 # parameters for conformer
"feature_dim": 80, "feature_dim": 80,
"subsampling_factor": 4, "subsampling_factor": 4,
"encoder_dim": 256, "encoder_dim": 512,
"nhead": 4, "nhead": 4,
"dim_feedforward": 1024, "dim_feedforward": 2048,
"num_encoder_layers": 18, "num_encoder_layers": 18,
# parameters for decoder # parameters for decoder
"decoder_dim": 512, "decoder_dim": 512,