From c367f1b819bb40e70e379f8f80b1d9fd67e47c79 Mon Sep 17 00:00:00 2001
From: zhuangzhong <zhuangzhong@corp.netease.com>
Date: 星期五, 07 六月 2024 16:15:16 +0800
Subject: [PATCH] add cif_wo_hidden_v1
---
funasr/train_utils/trainer_ds.py | 1 -
1 files changed, 0 insertions(+), 1 deletions(-)
diff --git a/funasr/train_utils/trainer_ds.py b/funasr/train_utils/trainer_ds.py
index 1a553f8..ec887cc 100644
--- a/funasr/train_utils/trainer_ds.py
+++ b/funasr/train_utils/trainer_ds.py
@@ -621,7 +621,6 @@
self.train_acc_avg = train_acc_avg.detach().cpu().item() / self.world_size
def forward_step(self, model, batch, loss_dict={}):
- dtype = torch.bfloat16
with maybe_autocast(dtype=self.dtype, use_deepspeed=self.use_deepspeed):
retval = model(**batch)
--
Gitblit v1.9.1