From 466a44f542ce7fe090d0bf47a21ebbadcd646c70 Mon Sep 17 00:00:00 2001 From: chenbohua3 Date: Sun, 18 Jul 2021 15:42:12 +0800 Subject: [PATCH] update docs --- nni/algorithms/compression/pytorch/quantization/quantizers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nni/algorithms/compression/pytorch/quantization/quantizers.py b/nni/algorithms/compression/pytorch/quantization/quantizers.py index 0e7ff11e962..09e53329b01 100644 --- a/nni/algorithms/compression/pytorch/quantization/quantizers.py +++ b/nni/algorithms/compression/pytorch/quantization/quantizers.py @@ -148,7 +148,7 @@ def __init__(self, model, config_list, optimizer=None, dummy_input=None): - dummy_input : tuple of tensor inputs to the model, which are used to get the graph of the module. The graph is used to find Conv-Bn patterns. And then the batch normalization folding would be enabled. If dummy_input is not - given, then batch normalization folding would be disabled. + given, the batch normalization folding would be disabled. """ super().__init__(model, config_list, optimizer, dummy_input)