提交 a4056d9c 编写于 作者: S SunAhong1993

modify graph_type

上级 bc871800
......@@ -59,7 +59,7 @@ class PaddleLayer(object):
class PaddleGraph(object):
def __init__(self, parent_layer=None, graph_type="dygraph"):
def __init__(self, parent_layer=None, graph_type="static"):
self.layers = OrderedDict()
self.edges_out = dict()
self.edges_in = dict()
......
......@@ -71,7 +71,7 @@ class PyTorchOpMapper(OpMapper):
current_node_outputs.extend(outputs)
# 初始化
graph = PaddleGraph(parent_layer)
graph = PaddleGraph(parent_layer, graph_type="dygraph")
current_node_outputs = []
graph_inputs = []
# 转换输入节点
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册