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)