提交 3e0a37d1 编写于 作者: S SunAhong1993

fix the interpolate

上级 4bb194dd
...@@ -3912,6 +3912,7 @@ def aten_upsample_bilinear2d(mapper, graph, node): ...@@ -3912,6 +3912,7 @@ def aten_upsample_bilinear2d(mapper, graph, node):
type="eq") type="eq")
layer_inputs["scale_factor"] = inputs_name[3] layer_inputs["scale_factor"] = inputs_name[3]
layer_attrs["align_mode"] = 0 layer_attrs["align_mode"] = 0
layer_attrs["mode"] = "bilinear"
graph.add_layer( graph.add_layer(
"paddle.nn.functional.interpolate", "paddle.nn.functional.interpolate",
inputs=layer_inputs, inputs=layer_inputs,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册