提交 9f034781 编写于 作者: D dongshuilong

fix slim bugs

上级 d5c76eee
...@@ -232,10 +232,10 @@ class Trainer_slim(Trainer): ...@@ -232,10 +232,10 @@ class Trainer_slim(Trainer):
self.config["Arch"]["name"], self.config["Arch"]["name"],
"best_model")) "best_model"))
elif self.config["Global"].get( elif self.config["Global"].get(
"pretraine_model", False) and os.path.exists(self.config[ "pretrained_model", False) and os.path.exists(self.config[
"Global"]["pretraine_model"] + ".pdparams"): "Global"]["pretrained_model"] + ".pdparams"):
load_dygraph_pretrain(self.model, load_dygraph_pretrain(self.model,
self.config["Global"]["pretraine_model"]) self.config["Global"]["pretrained_model"])
else: else:
raise RuntimeError( raise RuntimeError(
"The best_model or pretraine_model should exist to generate inference model" "The best_model or pretraine_model should exist to generate inference model"
......
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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册