From 4a82dcbb769530527249beaef1ee6eecb3388842 Mon Sep 17 00:00:00 2001 From: Alexander Dokuchaev Date: Tue, 31 Oct 2023 14:34:44 +0200 Subject: [PATCH] Update nncf/torch/quantization/quantize_model.py Co-authored-by: Daniil Lyakhov --- nncf/torch/quantization/quantize_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nncf/torch/quantization/quantize_model.py b/nncf/torch/quantization/quantize_model.py index 4f1ff46120c..d0d7619c709 100644 --- 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]: