diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/test_scaling_converter.py b/egs/librispeech/ASR/pruned_transducer_stateless3/test_scaling_converter.py index a9feea83c..2e131158f 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless3/test_scaling_converter.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless3/test_scaling_converter.py @@ -181,7 +181,7 @@ def test_convert_scaled_to_non_scaled(): y = torch.randint(low=1, high=vocab_size - 1, size=(N, U)) d1 = model.decoder(y) - d2 = model.decoder(y) + d2 = converted_model.decoder(y) assert torch.allclose(d1, d2)