Keras load_model command error : 'NoneType' object has no attribute 'get'

Please make sure that the boxes below are checked before you submit your issue. If your issue is an implementation question, please ask your question on StackOverflow or on the Keras Slack channel instead of opening a GitHub issue.

Thank you!

• Check that you are up-to-date with the master branch of Keras. You can update with: pip install git+git://github.com/keras-team/keras.git --upgrade --no-deps

• Check that your version of TensorFlow is up-to-date. The installation instructions can be found here.

• Provide a link to a GitHub Gist of a Python script that can reproduce your issue (or just copy the script here if it is short).

[ethanfowler]

In case it helps anyone else having this issue: my solution was to use the custom_objects argument of keras.model.load_model to provide a dictionary of custom objects/functions, including the function used within my Lambda layer. Model loads fine now.

In model definition:

x = tf.keras.layers.Lambda(self.lamba_function)(x)

Later, when loading the same model:

keras.models.load_model(filename, custom_objects={'lamba_function': self.lamba_function}

[robot010]

I’m having this issue after updating keras to 2.2.4. I have a python lambda function at my loss function, and using “load_model” with argument “custom_objects” won’t help. I was wondering if there is any other ways to deal with this issue besides only saving weights (discard the model structure).