TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.tpu.export_estimator_savedmodel

View source on GitHub

Export Estimator trained model for TPU inference.

tf.contrib.tpu.export_estimator_savedmodel(
    estimator,
    export_dir_base,
    serving_input_receiver_fn,
    assets_extra=None,
    as_text=False,
    checkpoint_path=None
)

Args:

  • estimator: Estimator with which model has been trained.
  • export_dir_base: A string containing a directory in which to create timestamped subdirectories containing exported SavedModels.
  • serving_input_receiver_fn: A function that takes no argument and returns a ServingInputReceiver or TensorServingInputReceiver.
  • assets_extra: A dict specifying how to populate the assets.extra directory within the exported SavedModel, or None if no extra assets are needed.
  • as_text: whether to write the SavedModel proto in text format.
  • checkpoint_path: The checkpoint path to export. If None (the default), the most recent checkpoint found within the model directory is chosen.

Returns:

The string path to the exported directory.