orbit.actions.ExportSavedModel

Action that exports the given model as a SavedModel.

orbit.actions.ExportSavedModel(
    model: tf.Module,
    file_manager: orbit.actions.ExportFileManager,
    signatures,
    options: Optional[tf.saved_model.SaveOptions] = None
)

model The model to export.
file_manager An instance of ExportFileManager (or a subclass), that provides file naming and cleanup functionality.
signatures The signatures to forward to tf.saved_model.save().
options Optional options to forward to tf.saved_model.save().

Methods

__call__

View source

__call__(
    _
)

Exports the SavedModel.