def METHOD_NAME(self):
    model = ImageClassifier(
        backbone=ResNet18V2Backbone(),
        num_classes=2,
    )
    model_output = model(self.input_batch)
    save_path = os.path.join(self.get_temp_dir(), "image_classifier.keras")
    model.save(save_path)
    restored_model = keras.models.load_model(save_path)
    # Check we got the real object back.
    self.assertIsInstance(restored_model, ImageClassifier)
    # Check that output matches.
    restored_output = restored_model(self.input_batch)
    self.assertAllClose(
        ops.convert_to_numpy(model_output),
        ops.convert_to_numpy(restored_output),
    )