conv batchnorm fuse

openvinotoolkit · Dec 18, 2024 · 7ee9fa2 · 7ee9fa2
1 parent 021cf2f
commit 7ee9fa2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/otx/core/exporter/native.py b/src/otx/core/exporter/native.py
@@ -68,6 +68,11 @@ def to_openvino(
         """
         dummy_tensor = torch.rand(self.input_size).to(next(model.parameters()).device)
 
+        # TODO(Eugene): Benchmark the performance of the model after conversion.
+        # for m in model.modules():
+        #     if hasattr(m, 'convert_to_deploy'):
+        #         m.convert_to_deploy()
+
         if self.via_onnx:
             with tempfile.TemporaryDirectory() as tmpdirname:
                 tmp_dir = Path(tmpdirname)