zhifu gao
2024-03-27 db82796f95e2f25e71d57df5f047c2354674c47a
funasr/train_utils/trainer.py
@@ -377,7 +377,8 @@
                    model=model,
                    dataloader_val=dataloader_val,
                    epoch=epoch,
                    writer=writer
                    writer=writer,
                    step=batch_idx+1,
                )
            if (batch_idx+1) % self.save_checkpoint_interval == 0: