From 1cd7e93183e41592f32444eef7625ec1e4081d19 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 5 Oct 2022 16:19:45 +0800 Subject: [PATCH] Fix bug setting layerdrop mask --- 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 11b18af35..011480a6c 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -362,7 +362,7 @@ class ConformerEncoder(nn.Module): mask[-final_layers_dropped:] = 0.0 layer_drop_prob = 0.075 - for i in range(final_layers_dropped): + for i in range(num_layers - final_layers_dropped): mask[i] = (rng.random() > layer_drop_prob) if mask.sum() == 0.0: mask[0] = 1.0