From 262388134d3b31dc6ec42fa15b99804d23e23d44 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sun, 27 Mar 2022 11:18:16 +0800 Subject: [PATCH] Increase model_warm_step to 4k --- egs/librispeech/ASR/pruned_transducer_stateless2/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index 13ba99017..c1e836903 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -296,7 +296,7 @@ def get_params() -> AttributeDict: "embedding_dim": 512, # parameters for Noam "warm_step": 60000, # For the 100h subset, use 8k - "model_warm_step": 3000, # arg given to model, not for lrate + "model_warm_step": 4000, # arg given to model, not for lrate "env_info": get_env_info(), } )