diff --git a/paddlespeech/s2t/utils/utility.py b/paddlespeech/s2t/utils/utility.py index fe2fa9167..d7e7c6ca2 100644 --- a/paddlespeech/s2t/utils/utility.py +++ b/paddlespeech/s2t/utils/utility.py @@ -131,8 +131,7 @@ def get_subsample(config): int: subsample rate. """ if config['encoder'] == 'squeezeformer': - input_layer = config["encoder_conf"]["time_reduction_layer_type"] - assert input_layer in ["conv2d", "conv1d", "stream"] + return 4 else: input_layer = config["encoder_conf"]["input_layer"] assert input_layer in ["conv2d", "conv2d6", "conv2d8"] @@ -142,7 +141,3 @@ def get_subsample(config): return 6 elif input_layer == "conv2d8": return 8 - elif input_layer == "conv1d": - return 6 - elif input_layer == "stream": - return 8