SavedModel format in TensorFlow - Practice Problems & Coding Challenges

Challenge - 5 Problems

🧠 Conceptual

intermediate

Understanding the SavedModel directory structure

Which of the following best describes the main components inside a TensorFlow SavedModel directory?

A A Python script that recreates the model architecture and loads weights from a checkpoint.

B A single '.h5' file that stores the entire model including architecture and weights.

C A JSON file describing the model architecture and a separate binary file for weights.

D A directory containing a 'saved_model.pb' file and a 'variables' folder with model weights.

💡 Hint

Think about what files TensorFlow saves when you use model.save('path') with the default format.

The SavedModel format saves the model as a directory containing a saved_model.pb file which holds the model architecture and computation graph, plus a variables folder that contains the trained weights.

❓ Predict Output

intermediate

Output of loading a SavedModel

What will be the output of the following code snippet?

import tensorflow as tf
model = tf.keras.Sequential([tf.keras.layers.Dense(1)])
model.save('my_model')
loaded_model = tf.keras.models.load_model('my_model')
print(type(loaded_model))

A <class 'tensorflow.keras.models.Sequential'>

B <class 'tensorflow.saved_model.LoadedModel'>

C AttributeError: module 'tensorflow.keras.models' has no attribute 'load_model'

D <class 'tensorflow.keras.models.Model'>

💡 Hint

Consider what type of object load_model returns when loading a Keras model saved in SavedModel format.

The load_model function returns a Keras Sequential object, which is of type tensorflow.keras.models.Sequential.

❓ Model Choice

advanced

Choosing the correct method to save a TensorFlow model in SavedModel format

You have a trained TensorFlow Keras model and want to save it in the SavedModel format for later use in TensorFlow Serving. Which method should you use?

A tf.saved_model.save(model, 'path_to_dir')

B model.save('path_to_dir', save_format='tf')

C model.save_weights('path_to_dir')

D tf.keras.models.save_model(model, 'path_to_dir', save_format='h5')

💡 Hint

Think about the recommended way to save a full Keras model including architecture and weights in SavedModel format.

Using model.save('path', save_format='tf') saves the entire model in SavedModel format, including architecture, weights, and optimizer state. The tf.saved_model.save function is lower level and requires a concrete function. Saving weights only does not save the architecture. Saving with save_format='h5' saves in HDF5 format, not SavedModel.

❓ Hyperparameter

advanced

Effect of including signatures when saving a SavedModel

When saving a TensorFlow model using tf.saved_model.save, what is the effect of specifying the signatures argument?

A It automatically tunes hyperparameters during saving for better performance.

B It compresses the SavedModel files to reduce disk space usage.

C It defines the input and output signatures for serving, enabling TensorFlow Serving to know how to call the model.

D It encrypts the SavedModel to protect the model weights.

💡 Hint

Think about why TensorFlow Serving needs information about inputs and outputs.

The signatures argument specifies the input and output formats (signatures) of the model functions. This helps TensorFlow Serving or other tools know how to call the model correctly.

🔧 Debug

expert

Diagnosing a loading error with a SavedModel

You saved a TensorFlow model using model.save('my_model'). Later, when you try to load it with tf.keras.models.load_model('my_model'), you get the error:

ValueError: No model found in config.

What is the most likely cause of this error?

A The model contains custom layers or objects but they were not provided during loading.

B The model was saved in HDF5 format but loaded as SavedModel format.

C The TensorFlow version used to load is older than the version used to save.

D The SavedModel directory is missing the 'variables' folder.

💡 Hint

Consider what happens if the model has custom components that TensorFlow does not know about by default.

If the model has custom layers or functions, you must provide them via the custom_objects argument when loading. Otherwise, TensorFlow cannot reconstruct the model architecture and raises this error.

Practice

1. What does the TensorFlow SavedModel format primarily store?

easy

A. Only the model's training data

B. Only the model's weights

C. Only the model's optimizer state

D. The complete model architecture, weights, and training configuration

Solution

1. Step 1: Understand what SavedModel stores

   The SavedModel format saves the entire model, including architecture, weights, and training configuration.

2. Step 2: Compare options with SavedModel features

   Options A, B, and D only mention parts of the model or incorrect elements, but SavedModel stores all parts together as in C.

3. Final Answer:

   The complete model architecture, weights, and training configuration -> Option D

4. Quick Check:

   SavedModel = full model save [OK]

Hint: SavedModel saves everything needed to reuse the model [OK]

Common Mistakes:

• Thinking SavedModel saves only weights
• Confusing SavedModel with checkpoint files
• Assuming training data is saved inside SavedModel

2. Which of the following is the correct way to save a TensorFlow Keras model in SavedModel format?

easy

A. model.save('path/to/model', save_format='tf')

B. model.save('path/to/model', save_format='h5')

C. tf.save_model(model, 'path/to/model')

D. model.save_weights('path/to/model')

Solution

1. Step 1: Recall the save_format options

   TensorFlow Keras models can be saved in 'h5' or 'tf' formats. 'tf' means SavedModel format.

2. Step 2: Check each option's correctness

   model.save('path/to/model', save_format='h5') saves in HDF5 format, not SavedModel. model.save('path/to/model', save_format='tf') correctly uses save_format='tf'. tf.save_model(model, 'path/to/model') is invalid syntax. model.save_weights('path/to/model') saves only weights, not full model.

3. Final Answer:

   model.save('path/to/model', save_format='tf') -> Option A

4. Quick Check:

   save_format='tf' means SavedModel [OK]

Hint: Use save_format='tf' to save in SavedModel format [OK]

Common Mistakes:

• Using save_format='h5' instead of 'tf'
• Trying to use tf.save_model which doesn't exist
• Saving only weights instead of full model

3. What will be the output of the following code snippet?

import tensorflow as tf
model = tf.keras.Sequential([tf.keras.layers.Dense(1)])
model.save('my_model')
loaded_model = tf.keras.models.load_model('my_model')
print(type(loaded_model))

medium

A. <class 'tensorflow.python.keras.engine.sequential.Sequential'>

B. <class 'tensorflow.python.keras.engine.functional.Functional'>

C. AttributeError

D. None

Solution

1. Step 1: Understand model saving and loading

   The model is a Sequential model saved in SavedModel format and then loaded back.

2. Step 2: Check the type of loaded model

   Loading a Sequential model returns an instance of tf.keras.Sequential class, so the type is Sequential.

3. Final Answer:

   <class 'tensorflow.python.keras.engine.sequential.Sequential'> -> Option A

4. Quick Check:

   Loaded model type matches original Sequential [OK]

Hint: Loaded model keeps original class type (Sequential here) [OK]

Common Mistakes:

• Expecting Functional model type after loading Sequential
• Thinking load_model returns None
• Confusing error with AttributeError

4. You saved a model using model.save('model_path') but when loading with tf.keras.models.load_model('wrong_path'), you get an error. What is the likely cause?

medium

A. The model was not compiled before saving

B. The model was saved in HDF5 format, so load_model cannot find SavedModel files

C. The load path is incorrect or does not exist

D. The model weights were not saved

Solution

1. Step 1: Analyze the error cause

   Loading a model from a wrong or non-existing path causes a file not found error.

2. Step 2: Check other options

   The model was saved in HDF5 format, so load_model cannot find SavedModel files is unrelated to path errors. The model was not compiled before saving and D do not cause loading errors but affect training or evaluation.

3. Final Answer:

   The load path is incorrect or does not exist -> Option C

4. Quick Check:

   Wrong path causes load_model error [OK]

Hint: Check file path carefully when loading models [OK]

Common Mistakes:

• Assuming model must be compiled before saving
• Confusing missing weights with loading errors
• Ignoring file path correctness

5. You want to deploy a TensorFlow model to a server. Which of these steps correctly uses the SavedModel format to save and load the model for deployment?

hard

A. Save with tf.saved_model.save(model, 'deploy_model') and load with tf.saved_model.load('deploy_model')

B. Save with model.save('deploy_model', save_format='tf') and load with tf.keras.models.load_model('deploy_model')

C. Save with model.save('deploy_model.h5') and load with tf.keras.models.load_model('deploy_model.h5')

D. Save with model.save_weights('deploy_model') and load with model.load_weights('deploy_model')

Solution

1. Step 1: Identify correct save and load methods for deployment

   Using model.save with save_format='tf' saves in SavedModel format, suitable for deployment. Loading with tf.keras.models.load_model restores the full model.

2. Step 2: Evaluate other options

   Save with model.save_weights('deploy_model') and load with model.load_weights('deploy_model') saves only weights, not full model. Save with model.save('deploy_model.h5') and load with tf.keras.models.load_model('deploy_model.h5') saves in HDF5, which is less preferred for deployment. Save with tf.saved_model.save(model, 'deploy_model') and load with tf.saved_model.load('deploy_model') uses low-level API which returns a different object, not a Keras model.

3. Final Answer:

   Save with model.save('deploy_model', save_format='tf') and load with tf.keras.models.load_model('deploy_model') -> Option B

4. Quick Check:

   Use model.save with save_format='tf' for deployment [OK]

Hint: Use model.save with save_format='tf' and load_model for deployment [OK]

Common Mistakes:

• Saving only weights for deployment
• Using tf.saved_model.load which returns non-Keras object
• Saving in HDF5 format instead of SavedModel