zhifu gao
2024-04-30 a09aba419f305abadc185ec41c336211549e894b
funasr/train_utils/trainer.py
@@ -456,7 +456,7 @@
                    batch_num_epoch = len(dataloader_train)
                self.log(
                    epoch,
                    batch_idx,
                    batch_idx + kwargs.get("start_step", 0),
                    step_in_epoch=self.step_in_epoch,
                    batch_num_epoch=batch_num_epoch,
                    lr=lr,