Update nncf/torch/quantization/quantize_model.py

Co-authored-by: Daniil Lyakhov <[email protected]>
openvinotoolkit · Oct 31, 2023 · 4a82dcb · 4a82dcb
1 parent 15f3204
commit 4a82dcb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nncf/torch/quantization/quantize_model.py b/nncf/torch/quantization/quantize_model.py
@@ -39,9 +39,9 @@ def create_nncf_network(model: torch.nn.Module, dataset: Dataset) -> NNCFNetwork
     Creates NNCFNetwork instance for the PyTorch model where the first item of dataset
     is used for model tracing.
 
-    :param model: PyTorch model
-    :param dataset: Dataset for model tracing
-    :return: NNCFNetwork instance for the input model
+    :param model: PyTorch model.
+    :param dataset: Dataset for model tracing.
+    :return: NNCFNetwork instance for the input model.
     """
 
     def get_inputs(dataloader_output: Any) -> Tuple[Tuple, Dict]: