Test

vshampor · Apr 13, 2023 · 683c224 · 683c224
1 parent 6321bfc
commit 683c224
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nncf/torch/model_creation.py b/nncf/torch/model_creation.py
@@ -53,7 +53,8 @@ def create_compressed_model(model: Module,
                             dummy_forward_fn: Callable[[Module], Any] = None,
                             wrap_inputs_fn: Callable[[Tuple, Dict], Tuple[Tuple, Dict]] = None,
                             wrap_outputs_fn: Callable[[Tuple, Dict], Tuple[Tuple, Dict]] = None,
-                            dump_graphs=True) \
+                            dump_graphs=True,
+                            new_arg=True) \
     -> Tuple[CompressionAlgorithmController, NNCFNetwork]:
     """
     The main function used to produce a model ready for compression fine-tuning from an original PyTorch