From 2186c6d61ae6ba381815b9f08f0496bbaafd5ecf Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Tue, 12 Jan 2021 21:22:25 +0800 Subject: [PATCH] fix --- x2paddle/op_mapper/dygraph/pytorch2paddle/aten.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/x2paddle/op_mapper/dygraph/pytorch2paddle/aten.py b/x2paddle/op_mapper/dygraph/pytorch2paddle/aten.py index 994980f..9dc06d3 100644 --- a/x2paddle/op_mapper/dygraph/pytorch2paddle/aten.py +++ b/x2paddle/op_mapper/dygraph/pytorch2paddle/aten.py @@ -4512,6 +4512,10 @@ def aten_upsample_bilinear2d(mapper, graph, node): current_outputs, scope_name) layer_inputs["align_corners"] = inputs_name[2] current_inputs.append(inputs_name[2]) + if "size" in layer_attrs and layer_attrs["size"] is None: + mapper._check_input(graph, inputs_node[3], inputs_name[3], + current_outputs, scope_name) + layer_inputs["scale_factor"] = inputs_name[3] layer_attrs["align_mode"] = 0 layer_attrs["mode"] = string("bilinear") graph.add_layer( @@ -4577,6 +4581,10 @@ def aten_upsample_nearest2d(mapper, graph, node): block = PaddleGraph(source_type="pytorch", parent_layer=if_layer, graph_type="dygraph") if_layer.add_block(block) if_layer.inputs["input-0"] = inputs_name[1] + if "size" in layer_attrs and layer_attrs["size"] is None: + mapper._check_input(graph, inputs_node[3], inputs_name[3], + current_outputs, scope_name) + layer_inputs["scale_factor"] = inputs_name[3] layer_attrs["align_mode"] = 0 layer_attrs["mode"] = string("nearest") graph.add_layer( -- GitLab