validate data

speech/cosyvoice/utils/executor.py

@@ -104,8 +104,8 @@ class Executor:
 
             if (
                 info_dict.get("save_per_step", -1) > 0
-                and (self.step + 1) % info_dict["save_per_step"] == 0
-                and (batch_idx + 1) % info_dict["accum_grad"] == 0
+                and (self.step) % info_dict["save_per_step"] == 0
+                and (batch_idx) % info_dict["accum_grad"] == 0
             ):
                 if dist.is_initialized():
                     dist.barrier()

speech/train.py

@@ -211,6 +211,7 @@ def main():
     start_step, start_epoch = 0, -1
     if args.checkpoint is not None:
         if os.path.exists(args.checkpoint):
+            logger.info(f"Load checkpoint from {args.checkpoint}")
             state_dict = torch.load(args.checkpoint, map_location="cpu")
             model.load_state_dict(state_dict, strict=False)
             if "step" in state_dict: