浏览代码

modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch

志浩 3 年之前
父节点
当前提交
b0abb2f4c0
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      funasr/models/pooling/statistic_pooling.py

+ 1 - 1
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 len(xs_pad.shape == 4):
+    if xs_pad.shape == 4:
         features = F.pad(xs_pad, (0, 0, pad, pad), "reflect")
     else:
         features = F.pad(xs_pad, (pad, pad), "reflect")