From 11023eea2182ad00f6a5acede6ec4182caf1fc07 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= <zhifu.gzf@alibaba-inc.com>
Date: Sat, 14 Sep 2024 14:47:32 +0800
Subject: [PATCH] speech2speech

---
 funasr/train_utils/trainer_ds.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/funasr/train_utils/trainer_ds.py b/funasr/train_utils/trainer_ds.py
index ff62811d8..04b84da2c 100644
--- a/funasr/train_utils/trainer_ds.py
+++ b/funasr/train_utils/trainer_ds.py
@@ -655,6 +655,8 @@ class Trainer:
             acc_log = loss_dict["stats"]["acc"].detach().cpu().item()
             if torch.isnan(loss_dict["loss"]):
                 logging.warning(f"loss is {loss_log}, set is to 0.0.\nitem:\n{batch['item']}")
+                if "item" in batch:
+                    logging.warning(f"item:\n{batch['item']}")
                 loss_log = 0.0
                 acc_log = 0.0