workerPoolSpecs:
  machineSpec:
    machineType: n1-standard-4
  replicaCount: 1
  pythonPackageSpec:
    executorImageUri: us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
    packageUris: gs://{BUCKET}/flowers-1.0.tar.gz  # TODO: Add your bucket!
    pythonModule: flowers.classifier.train
    args:
    - --job-dir="gs://{BUCKET}/flowers_trained/cpu"  # TODO: Add your bucket!
    - --pattern="-*"
    - --num_epochs=20
    - --distribute="cpu"