ds

游雁

2024-05-20 62acaf1158e9019cd49089d92175d47989bb1841

ds

 funasr/train_utils/trainer_ds.py

@@ -384,7 +384,7 @@
            if self.use_deepspeed:
                ckpt = os.path.join(self.output_dir, "model.pt")
                if os.path.exists(ckpt):
                    _, checkpoint = model_engine.load_checkpoint(self.output_dir, "model.pt")
                    _, checkpoint = model.load_checkpoint(self.output_dir, "model.pt")

                    self.saved_ckpts = checkpoint["saved_ckpts"]
                    self.val_acc_step_or_eoch = (