python-websocket funasr1.0 (#1310)

zhifu gao

2024-01-26 4f224c88068b66bcb6f81570da59d99c9bba8288

 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()