提交 ca3708c3 编写于 作者: S SunAhong1993

fix the bug

上级 661b2da3
......@@ -100,7 +100,7 @@ def main():
npy_weight = args.npy_path
fluid_model = args.model_param_path
outputs = None
if len(sys.argv) >= 6:
if args.need_layers_name:
outputs = args.need_layers_name.split(',')
ret = MyNet.convert(npy_weight, fluid_model, outputs)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册