remove

microsoft · Apr 16, 2021 · dc7f97b · dc7f97b
1 parent 6fdadc5
commit dc7f97b
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/nni/algorithms/compression/pytorch/quantization/quantizers.py b/nni/algorithms/compression/pytorch/quantization/quantizers.py
@@ -573,8 +573,6 @@ def __init__(self, model, config_list, optimizer=None):
         self.quant_grad = QuantForward()
         modules_to_compress = self.get_modules_to_compress()
         for layer, config in modules_to_compress:
-            # the way that lsq simulates quantization will make the gradient of zero point always
-            # equal to 0. So there may be no need to set it.
             layer.module.register_parameter("zero_point", torch.nn.Parameter(torch.Tensor([0.0])))
             layer.module.register_parameter("scale", torch.nn.Parameter(torch.Tensor([1.0])))
             if "weight" in config.get("quant_types", []):