From 235eb0746ff4b75a362c6eda11e47e5c57130519 Mon Sep 17 00:00:00 2001 From: kobenaxie <572745565@qq.com> Date: Sat, 27 Aug 2022 17:26:21 +0800 Subject: [PATCH] fix scaling converter test for decoder(predictor). (#553) --- .../ASR/pruned_transducer_stateless3/test_scaling_converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)