From d1a669162caf39fe15e318bfd3f51636cc8826bd Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 5 Apr 2022 13:31:52 +0800 Subject: [PATCH] Fix bug in lambda --- 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 45b3ca168..3b8f0499f 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -784,7 +784,7 @@ def run(rank, world_size, args): eps=1e-9, weight_decay=3e-04, target_rms=0.1) scheduler = torch.optim.lr_scheduler.LambdaLR( optimizer, - lambda step: (params.lr_num_steps/(step + params.lr_num_steps) ** params.lr_power)) + lambda step: ((params.lr_num_steps/(step + params.lr_num_steps)) ** params.lr_power)) if checkpoints and "optimizer" in checkpoints: