funasr/train_utils/trainer.py
@@ -103,6 +103,7 @@ os.makedirs(os.path.join(self.output_dir, "tensorboard"), exist_ok=True) self.writer = SummaryWriter(os.path.join(self.output_dir, "tensorboard")) if rank == 0 else None def _save_checkpoint(self, epoch, step=None):
