From b0f0c6c4ab369bc8f3222cbf20c5c134afd2cf2a Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 25 Jul 2022 04:38:12 +0800 Subject: [PATCH] Setting lr_update_period=(200,4k) in train.py --- egs/librispeech/ASR/pruned_transducer_stateless7/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py index 24829ae5b..21433dd59 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -928,7 +928,8 @@ def run(rank, world_size, args): optimizer = PrAdam(model.parameters(), lr=params.initial_lr, - max_block_size=512) + max_block_size=512, + lr_update_period=(200, 4000)) scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs)