Skip to content

Instantly share code, notes, and snippets.

@eileen-code4fun

eileen-code4fun/cifar10_config.yaml

Created May 26, 2021 14:45
Show Gist options
  • Save eileen-code4fun/5491caec2f16829ce52a2080aa0f0c39 to your computer and use it in GitHub Desktop.
Save eileen-code4fun/5491caec2f16829ce52a2080aa0f0c39 to your computer and use it in GitHub Desktop.
Download ZIP
CIFAR10 Config
Raw
cifar10_config.yaml
workerPoolSpecs:
machineSpec:
# Machines and GPUs: https://cloud.google.com/vertex-ai/docs/training/configure-compute#specifying_gpus
machineType: n1-standard-4
acceleratorType: NVIDIA_TESLA_V100
acceleratorCount: 2
replicaCount: 1
pythonPackageSpec:
# Executors: https://cloud.google.com/vertex-ai/docs/training/pre-built-containers
executorImageUri: us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
packageUris: GCS_PATH_FOR_PYTHON_CODE
pythonModule: trainer.task
# Train for 15 epochs.
args: --epochs=15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment