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
2 merge requests!251JVET-AI0230 AHG14: Fix for mismatch between encoding and decoding in the NNVC with palette on,!226Fix onnx 1.14
Pipeline #12817 passed
Stage: static-analysis
Stage: build
Hide whitespace changes
Inline Side-by-side
training/training_scripts/NN_Filtering/HOP4/model/model.py
+ 9
− 2
View file @ 6d5ddc39
...@@ -397,8 +397,15 @@ class SADLNet(nn.Sequential): ...@@ -397,8 +397,15 @@ class SADLNet(nn.Sequential):
model_for_export, model_for_export,
self.get_example_inputs(patch_size, batch_size), self.get_example_inputs(patch_size, batch_size),
filename, filename,
input_names=["in"], input_names=["in0", "in1", "in2", "in3", "in4", "in5"],
dynamic_axes={"in": {2: "h", 3: "w"}}, 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, opset_version=opset,
**kwargs, **kwargs,
) )
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment