提交 428edb6f 编写于 作者: G gaotingquan 提交者: Wei Shengyu

debug for infer

上级 c641fb3c
......@@ -43,6 +43,8 @@ class Engine(object):
assert mode in ["train", "eval", "infer", "export"]
self.mode = mode
self.config = config
self.eval_mode = self.config["Global"].get("eval_mode",
"classification")
# init logger
log_file = os.path.join(self.config['Global']['output_dir'],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册