Fix minimum input length for Conv2dSubsampling2 in check_short_utt

chintu619 · May 19, 2022 · bfe7bca · bfe7bca
1 parent 9396ed3
commit bfe7bca
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: