From d4fc6235b652ab60ce9d884c7011ab38bce26b24 Mon Sep 17 00:00:00 2001 From: chenbohua3 Date: Tue, 13 Jul 2021 11:06:37 +0800 Subject: [PATCH] rebase --- 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 cd0bddf9de..0229213233 100644 --- a/nni/algorithms/compression/pytorch/quantization/quantizers.py +++ b/nni/algorithms/compression/pytorch/quantization/quantizers.py @@ -171,7 +171,7 @@ def __init__(self, model, config_list, optimizer=None): self.bound_model.to(self.device) def validate_config(self, model, config_list): - schema = CompressorSchema([{ + schema = QuantizerSchema([{ Optional('quant_types'): Schema([lambda x: x in ['weight', 'output', 'input']]), Optional('quant_bits'): Or(And(int, lambda n: n == 8), Schema({ Optional('weight'): And(int, lambda n: n == 8),