Просмотр исходного кода

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

志浩 3 лет назад
Родитель
Сommit
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")