Fix nncf model export bug (#1346)

* Fix method name * Use public method for model_info Co-authored-by: Ashwin Vaidya <[email protected]>
openvinotoolkit · Nov 8, 2022 · 90ceb7c · 90ceb7c
1 parent 40b83d2
commit 90ceb7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/external/anomaly/tasks/nncf.py b/external/anomaly/tasks/nncf.py
@@ -201,7 +201,7 @@ def optimize(
 
         logger.info("Training completed.")
 
-    def _model_info(self) -> Dict:
+    def model_info(self) -> Dict:
         """Return model info to save the model weights.
 
         Returns: