speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

deepspeed

Browse Source
This commit is contained in:
游雁 2024-08-06 00:52:37 +08:00
parent 66411e2b5b
commit 6f48269f42
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/train_utils/trainer_ds.py
View File

@ -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