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

Update params

Browse Source
This commit is contained in:
Guanbo Wang 2022-04-21 02:30:54 -04:00
parent 2d07df5281
commit d6390fd107
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -317,7 +317,7 @@ def get_params() -> AttributeDict:
"batch_idx_train": 0, "batch_idx_train": 0,
"log_interval": 500, "log_interval": 500,
"reset_interval": 2000, "reset_interval": 2000,
"valid_interval": 3000, "valid_interval": 20000,
# parameters for conformer # parameters for conformer
"feature_dim": 80, "feature_dim": 80,
"subsampling_factor": 4, "subsampling_factor": 4,
@ -330,7 +330,7 @@ def get_params() -> AttributeDict:
# parameters for joiner # parameters for joiner
"joiner_dim": 512, "joiner_dim": 512,
# parameters for Noam # parameters for Noam
"model_warm_step": 10000, # arg given to model, not for lrate "model_warm_step": 20000, # arg given to model, not for lrate
"env_info": get_env_info(), "env_info": get_env_info(),
} }
) )