提交 56fb1e98 编写于 作者: W wjj19950828

add model.eval()

上级 8cef246f
......@@ -214,6 +214,7 @@ class ONNXConverter(object):
paddle.disable_static()
# run
model = paddle.jit.load(paddle_path)
model.eval()
result = model(*paddle_tensor_feed)
shutil.rmtree(
os.path.join(self.pwd, self.name, self.name + '_' + str(ver) +
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册