mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-18 21:44:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test code in pruned_transducer_stateless5/conformer.py

Browse Source
This commit is contained in:
yaozengwei 2022-07-07 19:51:27 +08:00
parent fef5e62ac4
commit f50198e66e
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
View File

@ -1267,7 +1267,6 @@ def _test_random_combine(final_weight: float, pure_prob: float, stddev: float):
num_channels = 50 num_channels = 50
m = RandomCombine( m = RandomCombine(
num_inputs=num_inputs, num_inputs=num_inputs,
num_channels=num_channels,
final_weight=final_weight, final_weight=final_weight,
pure_prob=pure_prob, pure_prob=pure_prob,
stddev=stddev, stddev=stddev,
@ -1290,7 +1289,7 @@ def _test_random_combine_main():
feature_dim = 50 feature_dim = 50
c = Conformer( 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 batch_size = 5
seq_len = 20 seq_len = 20