|TensorFlow 1 version||View source on GitHub|
Loads a model saved via
tf.keras.models.load_model( filepath, custom_objects=None, compile=True )
Used in the guide:
Used in the tutorials:
- Distributed training with Keras
- Save and load a model using a distribution strategy
- Save and load models
- Transfer learning with TensorFlow Hub
filepath: One of the following:
- String, path to the saved model
h5py.Fileobject from which to load the model
custom_objects: Optional dictionary mapping names (strings) to custom classes or functions to be considered during deserialization.
compile: Boolean, whether to compile the model after loading.
A Keras model instance. If an optimizer was found
as part of the saved model, the model is already
compiled. Otherwise, the model is uncompiled and
a warning will be displayed. When
compile is set
to False, the compilation is omitted without any
ImportError: if loading from an hdf5 file and h5py is not available.
IOError: In case of an invalid savefile.