From c9e048efef17672eff7d7eb1e336411d355be7d4 Mon Sep 17 00:00:00 2001
From: chenbohua3 <chen_bohua@hotmail.com>
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 6c8ff875ba4..d3c38eb682a 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),