From 2e0b2086586212a2c04b5eb612a85796be1494f7 Mon Sep 17 00:00:00 2001
From: Dr. Artificial曾小健 <875100501@qq.com>
Date: 星期二, 24 十二月 2024 13:46:31 +0800
Subject: [PATCH] typo fix (#2330)
---
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