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

auto frontend

Browse Source
This commit is contained in:
游雁 2024-06-06 19:48:40 +08:00
parent d6fedbfa77
commit 64941d99cc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/train_utils/trainer.py
View File

@ -481,7 +481,7 @@ class Trainer:
step_in_epoch=self.step_in_epoch, step_in_epoch=self.step_in_epoch,
batch_num_epoch=batch_num_epoch, batch_num_epoch=batch_num_epoch,
lr=lr, lr=lr,
loss=loss.detach().cpu().item(), loss=accum_grad * loss.detach().cpu().item(),
speed_stats=speed_stats, speed_stats=speed_stats,
stats=stats, stats=stats,
writer=writer, writer=writer,