funasr/train_utils/trainer.py @@ -149,6 +149,7 @@ self._train_epoch(epoch) if self.use_ddp or self.use_fsdp: dist.barrier() @@ -173,6 +174,7 @@ if self.use_ddp or self.use_fsdp: dist.barrier() if self.writer: self.writer.close()
