Skip to content

Commit 3fe4f67

Browse files
Enable output padding (#31)
* allow output padding * avoid non writeable np array warning * re-remove np.copy
1 parent 8438390 commit 3fe4f67

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

onnx2pytorch/convert/attribute.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,8 @@ def extract_attributes(node):
223223
kwargs["constant"] = extract_attr_values(attr)
224224
elif attr.name == "value_strings":
225225
kwargs["constant"] = extract_attr_values(attr)
226+
elif attr.name == "output_padding":
227+
kwargs["output_padding"] = extract_attr_values(attr)
226228
elif node.op_type == "Resize":
227229
# These parameters are not used, warn in Resize operator
228230
kwargs[attr.name] = extract_attr_values(attr)

0 commit comments

Comments
 (0)