Skip to content

Commit 0b34f3e

Browse files
authored
Merge pull request #406 from SunAhong1993/syf_dev
modify for v0.8.5
2 parents 5a20cb8 + 47ed21d commit 0b34f3e

File tree

3 files changed

+6
-5
lines changed

3 files changed

+6
-5
lines changed

x2paddle/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "0.8.4"
1+
__version__ = "0.8.5"
22

33
from .core.program import PaddleGraph
44

x2paddle/op_mapper/pytorch2paddle/aten.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ def aten_avg_pool2d(mapper, graph, node):
423423
graph.add_layer(
424424
"prim.assert",
425425
inputs={},
426-
outputs=[inputs_name[6]],
426+
outputs=[inputs_name[6] + "_assert"],
427427
type="eq",
428428
key=mapper.attrs[inputs_name[6]],
429429
value=None)
@@ -1473,15 +1473,15 @@ def aten_flatten(mapper, graph, node):
14731473
graph.add_layer(
14741474
"prim.assert",
14751475
inputs={},
1476-
outputs=[inputs_name[1]],
1476+
outputs=[inputs_name[1] + "_assert"],
14771477
type='eq',
14781478
key=mapper.attrs[inputs_name[1]],
14791479
value=1)
14801480
# 处理输入2,即%2
14811481
graph.add_layer(
14821482
"prim.assert",
14831483
inputs={},
1484-
outputs=[inputs_name[2]],
1484+
outputs=[inputs_name[2] + "_assert"],
14851485
type='eq',
14861486
key=mapper.attrs[inputs_name[2]],
14871487
value=-1)
@@ -2378,7 +2378,7 @@ def aten_max_pool2d(mapper, graph, node):
23782378
graph.add_layer(
23792379
"prim.assert",
23802380
inputs={},
2381-
outputs=[inputs_name[4]],
2381+
outputs=[inputs_name[4] + "_assert"],
23822382
type="eq",
23832383
key=mapper.attrs[inputs_name[4]],
23842384
value=[1, [1, 1]])
@@ -3912,6 +3912,7 @@ def aten_upsample_bilinear2d(mapper, graph, node):
39123912
type="eq")
39133913
layer_inputs["scale_factor"] = inputs_name[3]
39143914
layer_attrs["align_mode"] = 0
3915+
layer_attrs["mode"] = string("bilinear")
39153916
graph.add_layer(
39163917
"paddle.nn.functional.interpolate",
39173918
inputs=layer_inputs,

x2paddle/optimizer/pytorch_optimizer/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)