提交 a4056d9c 编写于 作者: S SunAhong1993

modify graph_type

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