CIFAR10 PL

cifar10_pl.py

from kfp import dsl
from google_cloud_pipeline_components import aiplatform

@dsl.pipeline(
  name='e2e-tutorial-automl-pipeline',
  description='A simple automl training',
  pipeline_root='GCS_PATH_FOR_PIPELINE'
)
def e2e_tutorial_pipeline():
  job = aiplatform.AutoMLImageTrainingJobRunOp(
    project='PROJECT_ID',
    display_name='e2e-tutorial-automl-pipeline',
    prediction_type='classification',
    model_type='CLOUD',
    dataset='DATASET_NAME',
    model_display_name='e2e-tutorial-automl-pipeline-model',
    budget_milli_node_hours=8000
  )