From 9f0347813573ba7620d1e923f1f270eead00dd8d Mon Sep 17 00:00:00 2001 From: dongshuilong Date: Mon, 9 Aug 2021 02:50:11 +0000 Subject: [PATCH] fix slim bugs --- deploy/slim/slim.py | 6 +++--- log/endpoints.log | 5 ----- 2 files changed, 3 insertions(+), 8 deletions(-) delete mode 100644 log/endpoints.log diff --git a/deploy/slim/slim.py b/deploy/slim/slim.py index f864fa5b7..f6940851c 100644 --- a/deploy/slim/slim.py +++ b/deploy/slim/slim.py @@ -232,10 +232,10 @@ class Trainer_slim(Trainer): self.config["Arch"]["name"], "best_model")) elif self.config["Global"].get( - "pretraine_model", False) and os.path.exists(self.config[ - "Global"]["pretraine_model"] + ".pdparams"): + "pretrained_model", False) and os.path.exists(self.config[ + "Global"]["pretrained_model"] + ".pdparams"): load_dygraph_pretrain(self.model, - self.config["Global"]["pretraine_model"]) + self.config["Global"]["pretrained_model"]) else: raise RuntimeError( "The best_model or pretraine_model should exist to generate inference model" diff --git a/log/endpoints.log b/log/endpoints.log deleted file mode 100644 index 438da7f1e..000000000 --- a/log/endpoints.log +++ /dev/null @@ -1,5 +0,0 @@ -PADDLE_TRAINER_ENDPOINTS: -127.0.0.1:51466 -127.0.0.1:58283 -127.0.0.1:34005 -127.0.0.1:58331 \ No newline at end of file