Skip to content

Commit 4275513

Browse files
[WA][conformance][TorchFX] reshape model explicitly (#3229)
Copy of the #3228 to the develop branch ### Changes TorchFX model in the conformance test are being reshaped explicitly before the serialization ### Reason for changes To work around dynamic shapes export OpenVINO bug 161495 ### Related tickets 161495 ### Tests job/weekly/job/Daemons/job/PTQ/job/ptq_single_model/7198 - passed
1 parent b0897b3 commit 4275513

File tree

1 file changed

+1
-0
lines changed
  • tests/post_training/pipelines

1 file changed

+1
-0
lines changed

tests/post_training/pipelines/base.py

+1
Original file line numberDiff line numberDiff line change
@@ -438,6 +438,7 @@ def save_compressed_model(self) -> None:
438438
elif self.backend == BackendType.FX_TORCH:
439439
exported_model = torch.export.export(self.compressed_model, (self.dummy_tensor,))
440440
ov_model = ov.convert_model(exported_model, example_input=self.dummy_tensor.cpu(), input=self.input_size)
441+
ov_model.reshape(self.input_size)
441442
self.path_compressed_ir = self.output_model_dir / "model.xml"
442443
ov.serialize(ov_model, self.path_compressed_ir)
443444
elif self.backend == BackendType.ONNX:

0 commit comments

Comments
 (0)