提交 f7a3cf03 编写于 作者: D Dun 提交者: qingqing01

fix (#1823)

上级 2346fd72
...@@ -51,13 +51,6 @@ def profile_context(profile=True): ...@@ -51,13 +51,6 @@ def profile_context(profile=True):
yield yield
def load_model(): def load_model():
load_vars = [
x for x in tp.list_vars()
if isinstance(x, fluid.framework.Parameter) and x.name.find('image_pool') ==
-1
]
fluid.io.load_vars(exe, dirname=args.init_weights_path, vars=load_vars)
return
if os.path.isdir(args.init_weights_path): if os.path.isdir(args.init_weights_path):
load_vars = [ load_vars = [
x for x in tp.list_vars() x for x in tp.list_vars()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册