Registration is open for TensorFlow Dev Summit 2020 Learn more

tfx.extensions.google_cloud_ai_platform.trainer.executor.Executor

View source on GitHub

Class Executor

Start a trainer job on Google Cloud AI Platform (GAIP).

Inherits From: BaseExecutor

__init__

View source

__init__(context=None)

Constructs a beam based executor.

Child Classes

class Context

Methods

Do

View source

Do(
    input_dict,
    output_dict,
    exec_properties
)

Starts a trainer job on Google Cloud AI Platform.

Args:

  • input_dict: Passthrough input dict for tfx.components.Trainer.executor.
  • output_dict: Passthrough input dict for tfx.components.Trainer.executor.
  • exec_properties: Mostly a passthrough input dict for tfx.components.Trainer.executor. custom_config.ai_platform_training_args is consumed by this class. For the full set of parameters supported by Google Cloud AI Platform, refer to https://cloud.google.com/ml-engine/docs/tensorflow/training-jobs#configuring_the_job

Returns:

None

Raises:

  • ValueError: if ai_platform_training_args is not in exec_properties.custom_config.
  • RuntimeError: if the Google Cloud AI Platform training job failed.