shixian.shi
2024-02-06 78ee057a99e8fb3c6ef9c0520a6bbd8c0f651cec
funasr/train_utils/trainer.py
@@ -148,6 +148,7 @@
            
            self._train_epoch(epoch)
            
            if self.use_ddp or self.use_fsdp:
                dist.barrier()
@@ -156,8 +157,8 @@
            if self.use_ddp or self.use_fsdp:
                dist.barrier()
            if self.rank == 0:
                self._save_checkpoint(epoch)
            
@@ -172,7 +173,8 @@
            
        if self.use_ddp or self.use_fsdp:
            dist.barrier()
        if self.writer:
            self.writer.close()