Merge pull request #78 from openspeech-team/ha

Resolved #76
openspeech-team · Aug 24, 2021 · 2976bad · 2976bad
2 parents a77aa3e + 2e037fe
commit 2976bad
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/openspeech/modules/conv2d_subsampling.py b/openspeech/modules/conv2d_subsampling.py
@@ -66,7 +66,6 @@ def __init__(
         )
 
     def forward(self, inputs: torch.Tensor, input_lengths: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor]:
-        outputs, input_lengths = super().forward(inputs, input_lengths)
-        output_lengths = input_lengths >> 2
-        output_lengths -= 1
+        outputs, output_lengths = super().forward(inputs, input_lengths)
+
         return outputs, output_lengths
diff --git a/tests/test_acoustic_models/test_conv2d_subsampling.py b/tests/test_acoustic_models/test_conv2d_subsampling.py
@@ -0,0 +1,16 @@
+import unittest
+import logging
+
+from openspeech.modules import Conv2dSubsampling
+from openspeech.utils import DUMMY_INPUTS, DUMMY_INPUT_LENGTHS
+
+logger = logging.getLogger(__name__)
+
+
+class TestConformer(unittest.TestCase):
+    def test_conv2d_subsampling(self):
+        conv_subsample = Conv2dSubsampling(input_dim=80, in_channels=1, out_channels=512)
+
+        outputs, output_lengths = conv_subsample(DUMMY_INPUTS, DUMMY_INPUT_LENGTHS)
+        print(f"input lengths : {DUMMY_INPUT_LENGTHS}")
+        print(f"output lengths : {output_lengths}")