Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

The Pusher TFX Pipeline Component

The Pusher component is used to push a validated model to a deployment target during model training or re-training. It relies on a ModelValidator component to ensure that the new model is "good enough" to be pushed to production.

  • Consumes: A Trained model in SavedModel format
  • Emits: The same SavedModel, along with versioning metadata

Using the Pusher Component

A Pusher pipeline component is typically very easy to deploy and requires little customization, since all of the work is done by the Pusher TFX component. Typical code looks like this:

from tfx import components

...

pusher = components.Pusher(
  model_export=trainer.outputs.output,
  model_blessing=model_validator.outputs.blessing,
  push_destination=pusher_pb2.PushDestination(
    filesystem=pusher_pb2.PushDestination.Filesystem(
        base_directory=serving_model_dir)
  )
)