Skip to content
Snippets Groups Projects
Commit 6d5ddc39 authored by Yun Li's avatar Yun Li
Browse files

Merge branch 'fix_onnx_1.14' into 'VTM-11.0_nnvc'

Fix onnx 1.14

See merge request !226
parents d7970f8d 8744ae9f
No related branches found
No related tags found
No related merge requests found
......@@ -397,8 +397,15 @@ class SADLNet(nn.Sequential):
model_for_export,
self.get_example_inputs(patch_size, batch_size),
filename,
input_names=["in"],
dynamic_axes={"in": {2: "h", 3: "w"}},
input_names=["in0", "in1", "in2", "in3", "in4", "in5"],
dynamic_axes={
"in0": {2: "h", 3: "w"},
"in1": {2: "h", 3: "w"},
"in2": {2: "h", 3: "w"},
"in3": {2: "h", 3: "w"},
"in4": {2: "h", 3: "w"},
"in5": {2: "h", 3: "w"},
},
opset_version=opset,
**kwargs,
)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment