修复chunk_size可能传入str或list结构 (#1357)

dudulu

2024-02-05 9c5f00794e7c474d1079ff78e5c69e8c53a30e6e

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