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":
