From 85c1675e7cf4ee036a1fdeded4da8426f1775d8a Mon Sep 17 00:00:00 2001
From: 凪咲 <36807427+chi2nagisa@users.noreply.github.com>
Date: 星期五, 19 七月 2024 10:26:58 +0800
Subject: [PATCH] fix: fix input download logic (#1929)
---
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