speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

speech2speech

Browse Source
This commit is contained in:
游雁 2024-09-14 14:47:32 +08:00
parent 21016fc693
commit 11023eea21
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/train_utils/trainer_ds.py
View File

@ -655,6 +655,8 @@ class Trainer:
acc_log = loss_dict["stats"]["acc"].detach().cpu().item() acc_log = loss_dict["stats"]["acc"].detach().cpu().item()
if torch.isnan(loss_dict["loss"]): if torch.isnan(loss_dict["loss"]):
logging.warning(f"loss is {loss_log}, set is to 0.0.\nitem:\n{batch['item']}") 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 loss_log = 0.0
acc_log = 0.0 acc_log = 0.0