mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-06 23:54:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Decrease model size, baseline is one Fangjun is running..

Browse Source
This commit is contained in:
Daniel Povey 2022-04-25 17:07:20 +08:00
parent aea116ea25
commit bbfa484196
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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