funasr/train_utils/trainer_ds.py @@ -272,6 +272,7 @@ ) else: print("Undo") if self.rank == 0: self.saved_ckpts[ckpt_name] = getattr( self, f"val_{self.avg_keep_nbest_models_type}_step_or_epoch" )[ckpt_name]