From fee3fd32fd2af544da864a20a73bb26e16aa2217 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期二, 14 三月 2023 10:58:25 +0800
Subject: [PATCH] add unit_test for speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch
---
funasr/datasets/large_datasets/utils/padding.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/datasets/large_datasets/utils/padding.py b/funasr/datasets/large_datasets/utils/padding.py
index 2e91e78..e814b1c 100644
--- a/funasr/datasets/large_datasets/utils/padding.py
+++ b/funasr/datasets/large_datasets/utils/padding.py
@@ -14,7 +14,7 @@
batch = {}
data_names = data[0].keys()
for data_name in data_names:
- if data_name == "key":
+ if data_name == "key" or data_name =="sampling_rate":
continue
else:
if data[0][data_name].dtype.kind == "i":
--
Gitblit v1.9.1