07_training/serverlessml/tpu_caip.yaml (
9
lines of code) (
raw
):
trainingInput: scaleTier: CUSTOM masterType: n1-standard-4 workerType: cloud_tpu workerCount: 1 workerConfig: acceleratorConfig: type: TPU_V2 count: 8