From 86845bd5d859ceb6f83cd83f3719c3e6641de987 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 8 Oct 2022 19:10:26 +0800 Subject: [PATCH] Change warmup schedule and increase warmup_batches from 4k to 6k --- egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index 8b1bf6ff9..210f89cc6 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -63,7 +63,7 @@ class Conformer(EncoderInterface): num_encoder_layers: Tuple[int] = (12, 12), dropout: float = 0.1, cnn_module_kernel: Tuple[int] = (31, 31), - warmup_batches: float = 4000.0, + warmup_batches: float = 6000.0, ) -> None: super(Conformer, self).__init__()