trainingInput:
  scaleTier: CUSTOM
  masterType: n1-standard-4
  workerType: cloud_tpu
  workerCount: 1
  workerConfig:
    acceleratorConfig:
      type: TPU_V2
      count: 8