fix slim bugs

2021-08-09 02:50:11 +00:00 · 2021-08-09 02:50:11 +00:00 · 9f03478135
parent d5c76eee56
commit 9f03478135
2 changed files with 3 additions and 8 deletions
--- 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"
--- a/log/endpoints.log
+++ b/log/endpoints.log
@ -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