From 6f48269f421785cba8435b3aee3e257e08975fba Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Tue, 6 Aug 2024 00:52:37 +0800 Subject: [PATCH] deepspeed --- funasr/train_utils/trainer_ds.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/train_utils/trainer_ds.py b/funasr/train_utils/trainer_ds.py index a0cd0df0d..9048a2f2d 100644 --- a/funasr/train_utils/trainer_ds.py +++ b/funasr/train_utils/trainer_ds.py @@ -231,7 +231,7 @@ class Trainer: model.save_checkpoint( save_dir=self.output_dir, tag=f"ds-model.pt", client_state=state ) - if not (step is None and epoch != 0): + if not (step is None and epoch == 0): if self.best_step_or_epoch == "": self.best_step_or_epoch = ckpt_name