From f50198e66ed9ed8f787dc26538b663055083933a Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Thu, 7 Jul 2022 19:51:27 +0800 Subject: [PATCH] fix test code in pruned_transducer_stateless5/conformer.py --- egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py index 49bc6a489..a30e97cf9 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py @@ -1267,7 +1267,6 @@ def _test_random_combine(final_weight: float, pure_prob: float, stddev: float): num_channels = 50 m = RandomCombine( num_inputs=num_inputs, - num_channels=num_channels, final_weight=final_weight, pure_prob=pure_prob, stddev=stddev, @@ -1290,7 +1289,7 @@ def _test_random_combine_main(): feature_dim = 50 c = Conformer( - num_features=feature_dim, output_dim=256, d_model=128, nhead=4 + num_features=feature_dim, d_model=128, nhead=4 ) batch_size = 5 seq_len = 20