tensorflow: KeyError: 'name' when try to load saved model with tf.keras.models.load_model()

Please make sure that this is a bug. As per our GitHub Policy, we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:bug_template

System information

• Have I written custom code (as opposed to using a stock example script provided in TensorFlow):
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Win10
• Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on a mobile device: None
• TensorFlow installed from (source or binary): Binary
• TensorFlow version (use command below): 2.2.0
• Python version: 3.7.7
• Bazel version (if compiling from source):
• GCC/Compiler version (if compiling from source):
• CUDA/cuDNN version: 10.1
• GPU model and memory: NVIDIA GeForce GTX 750 Ti 10GB (2GB Dedicated Memory; 8GB Shared Memory)

Describe the current behavior I keep getting an KeyError: 'name' when trying to load a saved model (.pb) with tf.keras.models.load_model(path)

Describe the expected behavior Load the model successfully.

Standalone code to reproduce the issue

Snippet where the error happens.

import tensorflow as tf 

# Load model
model = tf.keras.models.load_model("models/bs-32-ep-10-1595106737.3691123/") # Error happens here

# Do stuff
...

Other info / logs

My model is being saved within a class that inherits from tf.keras.models.Sequential(). I haven’t got any errors while training or saving my model.

def train(self):
        self.compile(
            loss=self.loss,
            optimizer=self.optimizer(lr=0.0001, decay=1e-6),
            metrics=[*self.mtcs]
        )

        self.fit(
            x=self.train_data[0],
            y=self.train_data[1],
            batch_size=self.batch_size,
            epochs=self.epochs,
            validation_data=(*self.test_data,),
            callbacks=[*self.callbacks]
        )

        if self.save_after_training:
            self.save("models/{}".format(self.NAME)) # Save Model

The Error:

Traceback (most recent call last):
  File "RNN.py", line 63, in <module>
    model = tf.keras.models.load_model("models/bs-32-ep-10-1595106737.3691123/")
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\saving\save.py", line 190, in load_model
    return saved_model_load.load(filepath, compile)
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\saving\saved_model\load.py", line 116, in load
    model = tf_load.load_internal(path, loader_cls=KerasObjectLoader)
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\saved_model\load.py", line 604, in load_internal
    export_dir)
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\saving\saved_model\load.py", line 188, in __init__
    super(KerasObjectLoader, self).__init__(*args, **kwargs)
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\saved_model\load.py", line 123, in __init__
    self._load_all()
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\saving\saved_model\load.py", line 215, in _load_all
    self._finalize_objects()
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\saving\saved_model\load.py", line 510, in _finalize_objects
    self._reconstruct_all_models()
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\saving\saved_model\load.py", line 528, in _reconstruct_all_models
    self._reconstruct_model(model_id, model, layers)
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\saving\saved_model\load.py", line 564, in _reconstruct_model
    config, created_layers={layer.name: layer for layer in layers})
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\engine\network.py", line 2019, in reconstruct_from_config
    process_layer(layer_data)
  File "C:\Users\EWC\github\GLAI_\venv\lib\site-packages\tensorflow\python\keras\engine\network.py", line 1993, in process_layer
    layer_name = layer_data['name']
KeyError: 'name'

If more code is required, please let me know. Thank you