python/FunASR-XL.git

FUNASR训练

blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace

游雁

2024-06-09 7b34f8ffde38c038c804c18208c3f2b129182949

 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,

			@@ -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,