From 6e69d784e4814c3dbe35e8f70c6cf4b920c8b20b Mon Sep 17 00:00:00 2001
From: 天地 <tiandiweizun@gmail.com>
Date: 星期三, 19 三月 2025 23:10:13 +0800
Subject: [PATCH] 1. bug fix:list(mean)和list(var)，由于mean和var是numpy，导致写入到文件的格式错误，参考上面的话，大概率是list(mean.tolist())，其实外层list没有必要 (#2437)
---
funasr/train_utils/device_funcs.py | 13 +++----------
1 files changed, 3 insertions(+), 10 deletions(-)
diff --git a/funasr/train_utils/device_funcs.py b/funasr/train_utils/device_funcs.py
index 7919e7d..dd7fc76 100644
--- a/funasr/train_utils/device_funcs.py
+++ b/funasr/train_utils/device_funcs.py
@@ -8,21 +8,14 @@
def to_device(data, device=None, dtype=None, non_blocking=False, copy=False):
"""Change the device of object recursively"""
if isinstance(data, dict):
- return {
- k: to_device(v, device, dtype, non_blocking, copy) for k, v in data.items()
- }
+ return {k: to_device(v, device, dtype, non_blocking, copy) for k, v in data.items()}
elif dataclasses.is_dataclass(data) and not isinstance(data, type):
return type(data)(
- *[
- to_device(v, device, dtype, non_blocking, copy)
- for v in dataclasses.astuple(data)
- ]
+ *[to_device(v, device, dtype, non_blocking, copy) for v in dataclasses.astuple(data)]
)
# maybe namedtuple. I don't know the correct way to judge namedtuple.
elif isinstance(data, tuple) and type(data) is not tuple:
- return type(data)(
- *[to_device(o, device, dtype, non_blocking, copy) for o in data]
- )
+ return type(data)(*[to_device(o, device, dtype, non_blocking, copy) for o in data])
elif isinstance(data, (list, tuple)):
return type(data)(to_device(v, device, dtype, non_blocking, copy) for v in data)
elif isinstance(data, np.ndarray):
--
Gitblit v1.9.1