提交 9672b749 编写于 作者: C chenguowei01

check model path

上级 8bc0d027
...@@ -111,6 +111,9 @@ def evaluate(cfg, ckpt_dir=None, use_gpu=False, use_mpio=False, **kwargs): ...@@ -111,6 +111,9 @@ def evaluate(cfg, ckpt_dir=None, use_gpu=False, use_mpio=False, **kwargs):
ckpt_dir = cfg.TEST.TEST_MODEL if not ckpt_dir else ckpt_dir ckpt_dir = cfg.TEST.TEST_MODEL if not ckpt_dir else ckpt_dir
if not os.path.exists(ckpt_dir):
raise ValueError('The ckpt_dir {} is not found'.format(ckpt_dir))
if ckpt_dir is not None: if ckpt_dir is not None:
print('load test model:', ckpt_dir) print('load test model:', ckpt_dir)
fluid.io.load_params(exe, ckpt_dir, main_program=test_prog) fluid.io.load_params(exe, ckpt_dir, main_program=test_prog)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册