From 4004ca81d84cda612265bd6919bea168e39601da Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 22 Mar 2022 13:32:24 +0800 Subject: [PATCH] Increase warm_step (and valid_interval) --- egs/librispeech/ASR/pruned_transducer_stateless2/train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index b9409127e..096f93d77 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -295,7 +295,7 @@ def get_params() -> AttributeDict: # parameters for decoder "embedding_dim": 512, # parameters for Noam - "warm_step": 30000, # For the 100h subset, use 8k + "warm_step": 60000, # For the 100h subset, use 8k "model_warm_step": 3000, # arg given to model, not for lrate "env_info": get_env_info(), } @@ -689,8 +689,8 @@ def run(rank, world_size, args): params = get_params() params.update(vars(args)) if params.full_libri is False: - params.valid_interval = 800 - params.warm_step = 16000 + params.valid_interval = 1600 + params.warm_step = 30000 fix_random_seed(params.seed) if world_size > 1: