提交 855cba84 编写于 作者: W wjj19950828

add run_dynamic switch

上级 41b51c34
......@@ -189,11 +189,11 @@ class ONNXConverter(object):
paddle_tensor_feed = list()
for i in range(len(self.input_feed)):
paddle_numpy_feed.append(self.input_feed[self.inputs_name[i]])
paddle_tensor_feed.append(paddle.to_tensor(self.input_feed[self.inputs_name[i]]))
paddle_tensor_feed.append(
paddle.to_tensor(self.input_feed[self.inputs_name[i]]))
if self.run_dynamic:
paddle_path = os.path.join(
self.pwd, self.name,
paddle_path = os.path.join(self.pwd, self.name,
self.name + '_' + str(ver) + '_paddle/')
import sys
sys.path.append(paddle_path)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册