From 7368b5891c07cb4be39ba35600447016b5bcbdf4 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 0e7ff11e96..09e53329b0 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)