From 8cac1c97d92a0cbfaf9862930404ede92150beeb Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期四, 04 五月 2023 19:22:56 +0800
Subject: [PATCH] update
---
funasr/datasets/large_datasets/utils/padding.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/funasr/datasets/large_datasets/utils/padding.py b/funasr/datasets/large_datasets/utils/padding.py
index fdca63d..7c66608 100644
--- a/funasr/datasets/large_datasets/utils/padding.py
+++ b/funasr/datasets/large_datasets/utils/padding.py
@@ -13,7 +13,8 @@
batch = {}
data_names = data[0].keys()
for data_name in data_names:
- if data_name == "key" or data_name =="sampling_rate":
+ if data_name == "key" or data_name == "sampling_rate" or data_name == 'hotword_indxs':
+ batch[data_name] = data[0][data_name]
continue
else:
if data[0][data_name].dtype.kind == "i":
--
Gitblit v1.9.1