未验证 提交 79a3dbc3 编写于 作者: L littletomatodonkey 提交者: GitHub

fix export effnet model (#451)

上级 27013344
...@@ -55,7 +55,10 @@ def create_model(args, model, input, class_dim=1000): ...@@ -55,7 +55,10 @@ def create_model(args, model, input, class_dim=1000):
def main(): def main():
args = parse_args() args = parse_args()
model = architectures.__dict__[args.model]() if "EfficientNet" in args.model:
model = architectures.__dict__[args.model](is_test=True)
else:
model = architectures.__dict__[args.model]()
place = fluid.CPUPlace() place = fluid.CPUPlace()
exe = fluid.Executor(place) exe = fluid.Executor(place)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册