export for default (#1883)

c5c0e72a · Cjkkkk · chicm-ms · 9f40659d · c5c0e72a
Commit c5c0e72a authored Dec 27, 2019 by Cjkkkk Committed by chicm-ms Dec 27, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/sdk/pynni/nni/compression/torch/builtin_quantizers.py src/sdk/pynni/nni/compression/torch/builtin_quantizers.py +1 -1

No files found.
--- a/src/sdk/pynni/nni/compression/torch/builtin_quantizers.py
+++ b/src/sdk/pynni/nni/compression/torch/builtin_quantizers.py
@@ -5,7 +5,7 @@ import logging
 import torch
 from .compressor import Quantizer, QuantGrad, QuantType

-__all__ = ['NaiveQuantizer', 'QAT_Quantizer', 'DoReFaQuantizer']
+__all__ = ['NaiveQuantizer', 'QAT_Quantizer', 'DoReFaQuantizer', 'BNNQuantizer']

 logger = logging.getLogger(__name__)