def METHOD_NAME(self):
    model = EfficientNetV1B0Backbone()
    model_output = model(self.input_batch)
    save_path = os.path.join(
        self.get_temp_dir(), "efficientnet_v1_backbone.keras"
    )
    model.save(save_path)
    restored_model = keras.models.load_model(save_path)
    # Check we got the real object back.
    # Note that these aliases serialized as the base class
    self.assertIsInstance(restored_model, EfficientNetV1Backbone)
    # Check that output matches.
    restored_output = restored_model(self.input_batch)
    self.assertAllClose(model_output, restored_output)