From d50edc297a7e6d944a244a84adf68b8bcce5a057 Mon Sep 17 00:00:00 2001
From: nsdou <168500039+nsdou@users.noreply.github.com>
Date: 星期三, 15 五月 2024 17:48:37 +0800
Subject: [PATCH] Fix incorrect assignment of 'end' attribute to 'start' in sentences list comprehension (#1680)
---
funasr/datasets/large_datasets/utils/clipping.py | 8 ++++++--
1 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/funasr/datasets/large_datasets/utils/clipping.py b/funasr/datasets/large_datasets/utils/clipping.py
index 2554aba..92f7d70 100644
--- a/funasr/datasets/large_datasets/utils/clipping.py
+++ b/funasr/datasets/large_datasets/utils/clipping.py
@@ -25,7 +25,9 @@
tensor_lengths = torch.tensor([len(d[data_name]) for d in data], dtype=torch.int32)
length_clip = min(tensor_lengths)
- tensor_clip = tensor_list[0].new_zeros(len(tensor_list), length_clip, tensor_list[0].shape[1])
+ tensor_clip = tensor_list[0].new_zeros(
+ len(tensor_list), length_clip, tensor_list[0].shape[1]
+ )
for i, (tensor, length) in enumerate(zip(tensor_list, tensor_lengths)):
diff = length - length_clip
assert diff >= 0
@@ -35,6 +37,8 @@
tensor_clip[i] = crop_to_max_size(tensor, length_clip)
batch[data_name] = tensor_clip
- batch[data_name + "_lengths"] = torch.tensor([tensor.shape[0] for tensor in tensor_clip], dtype=torch.long)
+ batch[data_name + "_lengths"] = torch.tensor(
+ [tensor.shape[0] for tensor in tensor_clip], dtype=torch.long
+ )
return keys, batch
--
Gitblit v1.9.1