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

fix slim bugs

上级 d5c76eee
......@@ -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"
......
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.
先完成此消息的编辑!
想要评论请 注册