| funasr/datasets/large_datasets/utils/padding.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
funasr/datasets/large_datasets/utils/padding.py
@@ -13,10 +13,10 @@ batch = {} data_names = data[0].keys() for data_name in data_names: if data_name == "key" or data_name == "sampling_rate" or data_name == 'hotword_indxs': batch[data_name] = data[0][data_name] if data_name == "key" or data_name == "sampling_rate": continue else: if data_name != 'hotword_indxs': if data[0][data_name].dtype.kind == "i": pad_value = int_pad_value tensor_type = torch.int64