From 3dd25d6b2d60e9234f461cbdb06654ffd3f592d3 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 3 Apr 2023 12:13:09 +0800 Subject: [PATCH] Increase feature_mask_dropout_prob to 0.125 --- egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index 18b58da92..866e0a174 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -306,7 +306,7 @@ class Zipformer2(EncoderInterface): # we divide the dropped-out feature dimensions into two equal groups; # the first group is dropped out with probability 0.1, the second # with probability approximately twice that. - feature_mask_dropout_prob = 0.1 + feature_mask_dropout_prob = 0.125 # frame_mask_max1 shape: (num_frames_max, batch_size, 1) frame_mask_max1 = (torch.rand(num_frames_max, batch_size, 1,