Calling submodels in train_step throws ValueError when saving and loading error.

[hubgrim]

Calling submodels in train_step throws ValueError when saving and loading error.

[sushreebarsa]

@hubgrim I am able to get a different error while reproducing the issue reported here. Kindly check the gist for reference.
The .keras extension typically represents the recommended Keras v3 format. If your model was saved in a different format (e.g., HDF5 - .h5), you'll need to use the appropriate loading function as follows;

loaded_model = tf.keras.models.load_model('custom_model.h5')  # For HDF5 format

Thank you!

[github-actions]

This issue is stale because it has been open for 7 days with no activity. It will be closed if no further activity occurs. Thank you.

[hubgrim]

@hubgrim I am able to get a different error while reproducing the issue reported here. Kindly check the gist for reference. The .keras extension typically represents the recommended Keras v3 format. If your model was saved in a different format (e.g., HDF5 - .h5), you'll need to use the appropriate loading function as follows;

loaded_model = tf.keras.models.load_model('custom_model.h5')  # For HDF5 format

Thank you!

Hm, I don't know why this error differs or why the saving format would be an issue, since I use the .keras extension when loading and saving. The lines are right next to each other.

[sushreebarsa]

@hubgrim While the .keras extension suggests a Keras model, there may be compatibility issues between the version of Keras you used to save the model and the version you're using to load it. Even small version bumps can sometimes cause errors. Please ensure the version compatibility.
Thank you!

[hubgrim]

The error arises even when saving/loading within the same script. I did not save the model at some point and reloaded it at some other time, but within the same call of the program. Both the model, that provides .save(), and load_model() were imported from tensorflow.keras.models so I don't see how this could be rooted in version incompatibility.