Merge pull request espnet#4378 from akreal/fix-check_short_utt

Fix minimum input length for Conv2dSubsampling2 in check_short_utt
chintu619 · May 19, 2022 · 047d0c4 · 047d0c4
2 parents 9396ed3 + bfe7bca
commit 047d0c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/espnet/nets/pytorch_backend/transformer/subsampling.py b/espnet/nets/pytorch_backend/transformer/subsampling.py
@@ -30,8 +30,8 @@ def __init__(self, message, actual_size, limit):
 
 def check_short_utt(ins, size):
     """Check if the utterance is too short for subsampling."""
-    if isinstance(ins, Conv2dSubsampling2) and size < 3:
-        return True, 3
+    if isinstance(ins, Conv2dSubsampling2) and size < 7:
+        return True, 7
     if isinstance(ins, Conv2dSubsampling) and size < 7:
         return True, 7
     if isinstance(ins, Conv2dSubsampling6) and size < 11: