10d7ece1c34c5533262a54d20d974a3d04f0b7ce,src/sdk/pynni/nni/compression/tensorflow/compressor.py,Compressor,__init__,#Compressor#Any#Any#Any#,79

Before Change

    def __init__(self, LayerWrapperClass, model, config_list):
        assert isinstance(model, tf.keras.Model)
        if isinstance(model, tf.keras.Sequential):
            raise ValueError("NNI model compression does not support `Sequential` model for now")
        self.validate_config(model, config_list)

        self.bound_model = model
        self.wrappers = []

After Change

        self._original_model = model
        self._config_list = config_list
        self._wrapper_class = LayerWrapperClass
        self._wrappers = {}  // key: id(layer) , value: Wrapper(layer)

        self.compressed_model = self._instrument(model)
        self.wrappers = list(self._wrappers.values())

        if not self.wrappers:
            _logger.warning("Nothing is configured to compress, please check your model and config list")

Instances