From 28ccfbfc51068a663a80764e14074df5edf2b5ba Mon Sep 17 00:00:00 2001
From: kongdeqiang <kongdeqiang960204@163.com>
Date: 星期五, 13 三月 2026 17:41:41 +0800
Subject: [PATCH] 提交
---
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