From 69f733c89ba5191ee12e821c2510f62e6190256e Mon Sep 17 00:00:00 2001 From: Ping Zheng Date: Thu, 4 Jul 2024 11:45:54 -0400 Subject: [PATCH] fix --- tests/test_conformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_conformer.py b/tests/test_conformer.py index 0bd25424..29693c82 100644 --- a/tests/test_conformer.py +++ b/tests/test_conformer.py @@ -37,7 +37,7 @@ def get_output_shape(batch, time, features, norm=None, kernel_size=31, dropout=0 def test_ConformerPositionwiseFeedForwardV1(): def get_output_shape(input_shape, input_dim, hidden_dim, dropout, activation): x = torch.randn(input_shape) - cfg = ConformerPositionwiseFeedForwardV2Config( + cfg = ConformerPositionwiseFeedForwardV1Config( input_dim=input_dim, hidden_dim=hidden_dim, dropout=dropout, activation=activation ) conf_ffn_part = ConformerPositionwiseFeedForwardV1(cfg)