From 0423b067da519b435d7ca6cb2c484aeeb085fbcd Mon Sep 17 00:00:00 2001 From: wjj19950828 Date: Wed, 2 Nov 2022 15:20:00 +0800 Subject: [PATCH] Support GFPGAN --- x2paddle/op_mapper/pytorch2paddle/aten.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/x2paddle/op_mapper/pytorch2paddle/aten.py b/x2paddle/op_mapper/pytorch2paddle/aten.py index b393edf..00229c0 100755 --- a/x2paddle/op_mapper/pytorch2paddle/aten.py +++ b/x2paddle/op_mapper/pytorch2paddle/aten.py @@ -4501,10 +4501,10 @@ def aten_rand(mapper, graph, node): layer_inputs = {} layer_attrs = {} inputs_name, inputs_node = mapper._get_inputs_name(node) - # 获取当前节点输出的list + # outputs list current_outputs = [output_name] current_inputs = [] - # 处理输入0，即%23，代表end + # deal with shape if inputs_name[0] in mapper.attrs: layer_attrs["shape"] = mapper.attrs[inputs_name[0]] else: @@ -4512,7 +4512,7 @@ def aten_rand(mapper, graph, node): current_outputs, scope_name) layer_inputs["shape"] = inputs_name[0] current_inputs.append(inputs_name[0]) - # 处理输入1，即%8，代表dtype + # deal with dtype layer_attrs["dtype"] = dtype_dict[mapper.attrs[inputs_name[1]]] graph.add_layer( @@ -4542,10 +4542,10 @@ def aten_randn(mapper, graph, node): layer_inputs = {} layer_attrs = {} inputs_name, inputs_node = mapper._get_inputs_name(node) - # 获取当前节点输出的list + # outputs list current_outputs = [output_name] current_inputs = [] - # 处理输入0，即%23，代表end + # deal with shape if inputs_name[0] in mapper.attrs: layer_attrs["shape"] = mapper.attrs[inputs_name[0]] else: @@ -4553,7 +4553,7 @@ def aten_randn(mapper, graph, node): current_outputs, scope_name) layer_inputs["shape"] = inputs_name[0] current_inputs.append(inputs_name[0]) - # 处理输入1，即%8，代表dtype + # deal with dtype layer_attrs["dtype"] = dtype_dict[mapper.attrs[inputs_name[1]]] graph.add_layer( -- GitLab