From 5e7a8d1ccae80e54f2e2ecfffdf8e4294800b5c3 Mon Sep 17 00:00:00 2001
From: Legend <me@liux.pro>
Date: 星期日, 15 十二月 2024 01:47:12 +0800
Subject: [PATCH] Update readme_zh.md (#2312)
---
funasr/train_utils/trainer_ds.py | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/funasr/train_utils/trainer_ds.py b/funasr/train_utils/trainer_ds.py
index 8a0679c..85513a5 100644
--- a/funasr/train_utils/trainer_ds.py
+++ b/funasr/train_utils/trainer_ds.py
@@ -30,8 +30,9 @@
yield
else:
if dtype == torch.float16 or dtype == torch.bfloat16:
- with autocast(enabled=True, dtype=dtype):
- yield
+ yield
+ # with autocast(enabled=True, dtype=dtype):
+ # yield
else:
yield
@@ -477,7 +478,7 @@
for k_ex in self.excludes:
k_tmp = k.replace("module.", "")
if k_tmp.startswith(k_ex):
- logging.info(f"key: {{k}} matching: {k_ex}, excluded")
+ logging.info(f"key: {k} matching: {k_ex}, excluded")
excludes_flag = True
break
if excludes_flag:
--
Gitblit v1.9.1