From f191f4c868af7ca73e4fae6242339309ae15d88c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BF=97=E6=B5=A9?= Date: Tue, 14 Mar 2023 14:31:27 +0800 Subject: [PATCH] modify statistic pooling layer --- funasr/models/pooling/statistic_pooling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/models/pooling/statistic_pooling.py b/funasr/models/pooling/statistic_pooling.py index 97f8a24f5..8f85de99d 100644 --- a/funasr/models/pooling/statistic_pooling.py +++ b/funasr/models/pooling/statistic_pooling.py @@ -82,7 +82,7 @@ def windowed_statistic_pooling( tt = xs_pad.shape[2] num_chunk = int(math.ceil(tt / pooling_stride)) pad = pooling_size // 2 - if xs_pad.shape == 4: + if len(xs_pad.shape) == 4: features = F.pad(xs_pad, (0, 0, pad, pad), "reflect") else: features = F.pad(xs_pad, (pad, pad), "reflect")