10d7ece1c34c5533262a54d20d974a3d04f0b7ce,src/sdk/pynni/nni/compression/tensorflow/compressor.py,,_instrument_model,#Any#Any#,289

Before Change

                name, index = key
                cur = getattr(cur, name)[index]
        key = wrapper.layer_info.path[-1]
        if isinstance(key, str):
            setattr(cur, key, wrapper)
        else:
            name, index = key
            getattr(cur, name)[index] = wrapper
            //if isinstance(cur, tf.keras.Sequential):
            //    cur._graph_initialized = False
            //    cur._layer_call_argspecs[wrapper] = cur._layer_call_argspecs[wrapper.layer]

After Change

        return tf.keras.Sequential(layers) if need_rebuild else seq

    def _instrument_model(self, model):
        for key, value in list(model.__dict__.items()):  // avoid "dictionary keys changed during iteration"
            if isinstance(value, tf.keras.layers.Layer):
                new_layer = self._instrument(value)
                if new_layer is not value:

Instances