update

Author: andrey-churkin

nncf/tensorflow/quantization/quantize_model.py

@@ -184,7 +184,7 @@ def quantize_impl(
     # NOTE: We set the config here to properly save/load the quantized model during training into tf.train.Checkpoint.
     # You can obtain that config via the nncf.tensorflow.get_config() method and save/load it to/from
     # tf.train.Checkpoint using the nncf.tensorflow.ConfigState class.
-    config = compression_ctrl.get_compression_state()["builder_state"]["quantization"]
+    config = compression_ctrl.get_compression_state()["builder_state"]
     setattr(compressed_model, "_nncf_config", config)
 
     return compressed_model

nncf/tensorflow/utils/state.py

@@ -106,9 +106,10 @@ def serialize(self) -> str:
 
         :return: A serialized config.
         """
+        quantizer_setup_state = self.config["quantization"]["quantizer_setup"]
         data = {
             "quantization": {
-                "quantizer_setup": TFQuantizationSetup.from_state(self.config["quantizer_setup"]).get_state(),
+                "quantizer_setup": TFQuantizationSetup.from_state(quantizer_setup_state).get_state(),
             }
         }
         return json.dumps(data)