Add Yolov8

openvinotoolkit · Jul 12, 2024 · 42deabb · 42deabb
1 parent 9dc4239
commit 42deabb
Show file tree

Hide file tree

Showing 4 changed files with 1,596 additions and 58 deletions.
diff --git a/nncf/common/utils/backend.py b/nncf/common/utils/backend.py
@@ -58,8 +58,8 @@ def is_torch_model(model: TModel) -> bool:
     :param model: A target model.
     :return: True if the model is an instance of torch.nn.Module and not torch.fx.GraphModule, otherwise False.
     """
-    import torch #type: ignore
-    import torch.fx #type: ignore
+    import torch  # type: ignore
+    import torch.fx  # type: ignore
 
     return not isinstance(model, torch.fx.GraphModule) and isinstance(model, torch.nn.Module)
 
@@ -71,7 +71,7 @@ def is_torch_fx_model(model: TModel) -> bool:
     :param model: A target model.
     :return: True if the model is an instance of torch.fx.GraphModule, otherwise False.
     """
-    import torch.fx #type: ignore
+    import torch.fx
 
     return isinstance(model, torch.fx.GraphModule)