zhong zhuang
2024-06-14 145022c438c8158e6bffdc5ff66fdd28468a0a18
funasr/train_utils/trainer.py
@@ -481,7 +481,7 @@
                    step_in_epoch=self.step_in_epoch,
                    batch_num_epoch=batch_num_epoch,
                    lr=lr,
                    loss=loss.detach().cpu().item(),
                    loss=accum_grad * loss.detach().cpu().item(),
                    speed_stats=speed_stats,
                    stats=stats,
                    writer=writer,