mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-28 11:14:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reduce the number of decoder layers from 4 to 2.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-12-18 11:25:33 +08:00
parent 9d0d5d19fb
commit d362a3dba7
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/transducer/decode.py
View File

@ -131,7 +131,7 @@ def get_params() -> AttributeDict:
"vgg_frontend": False, "vgg_frontend": False,
# decoder params # decoder params
"decoder_embedding_dim": 1024, "decoder_embedding_dim": 1024,
"num_decoder_layers": 4, "num_decoder_layers": 2,
"decoder_hidden_dim": 512, "decoder_hidden_dim": 512,
"env_info": get_env_info(), "env_info": get_env_info(),
} }

2
egs/librispeech/ASR/transducer/export.py
View File

@ -121,7 +121,7 @@ def get_params() -> AttributeDict:
"vgg_frontend": False, "vgg_frontend": False,
# decoder params # decoder params
"decoder_embedding_dim": 1024, "decoder_embedding_dim": 1024,
"num_decoder_layers": 4, "num_decoder_layers": 2,
"decoder_hidden_dim": 512, "decoder_hidden_dim": 512,
"env_info": get_env_info(), "env_info": get_env_info(),
} }

2
egs/librispeech/ASR/transducer/pretrained.py
View File

@ -118,7 +118,7 @@ def get_params() -> AttributeDict:
"vgg_frontend": False, "vgg_frontend": False,
# decoder params # decoder params
"decoder_embedding_dim": 1024, "decoder_embedding_dim": 1024,
"num_decoder_layers": 4, "num_decoder_layers": 2,
"decoder_hidden_dim": 512, "decoder_hidden_dim": 512,
"env_info": get_env_info(), "env_info": get_env_info(),
} }

2
egs/librispeech/ASR/transducer/train.py
View File

@ -198,7 +198,7 @@ def get_params() -> AttributeDict:
"vgg_frontend": False, "vgg_frontend": False,
# decoder params # decoder params
"decoder_embedding_dim": 1024, "decoder_embedding_dim": 1024,
"num_decoder_layers": 4, "num_decoder_layers": 2,
"decoder_hidden_dim": 512, "decoder_hidden_dim": 512,
# parameters for Noam # parameters for Noam
"warm_step": 80000, # For the 100h subset, use 8k "warm_step": 80000, # For the 100h subset, use 8k