From bb233d3449a134e0d448ef277a71fb922b39ba57 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 5 Oct 2022 23:18:50 +0800 Subject: [PATCH] Add debug info --- egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index 011480a6c..c177b1d78 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -371,6 +371,8 @@ class ConformerEncoder(nn.Module): mask = get_random_mask() device = self.to_layerdrop_scales[0].weight.device layerdrop_scales = 1.0 + self.to_layerdrop_scales(mask.to(device)) + if random.random() < 0.05: + logging.info(f"mask={mask}, layerdrop_scales = {layerdrop_scales.to('cpu')}") return mask, layerdrop_scales