diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 83a24a601dc4361f8d6d178b862e0b691d30b960..af4fbfeb7179b7824b74f8e92420f9639163ad79 100644 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -308,14 +308,14 @@ class Engine(object): self.output_dir, model_name=self.config["Arch"]["name"], prefix="epoch_{}".format(epoch_id)) - # save the latest model - save_load.save_model( - self.model, - self.optimizer, {"metric": acc, - "epoch": epoch_id}, - self.output_dir, - model_name=self.config["Arch"]["name"], - prefix="latest") + # save the latest model + save_load.save_model( + self.model, + self.optimizer, {"metric": acc, + "epoch": epoch_id}, + self.output_dir, + model_name=self.config["Arch"]["name"], + prefix="latest") if self.vdl_writer is not None: self.vdl_writer.close()