From 0f8ee68af22657c90c5e2762a5e48e1d09b7ce0c Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 8 Apr 2022 16:53:42 +0800 Subject: [PATCH] Fix bug --- egs/librispeech/ASR/pruned_transducer_stateless2/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index d5da5d0e9..038469282 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -784,6 +784,7 @@ def run(rank, world_size, args): # The `epoch` variable in the lambda expression binds to the value below # in `for epoch in range(params.start_epoch, params.num_epochs):`. + epoch = 0 scheduler = torch.optim.lr_scheduler.LambdaLR( optimizer, lambda step: (((step + params.lr_begin_steps) / params.lr_begin_steps) ** -0.5 *