CIFAR10 Viz Request

cifar10_viz_request.json

{
  "displayName": "e2e-tutorial-viz",
  "jobSpec": {
    "workerPoolSpecs": [
      {
        "replicaCount": 1,
        "machineSpec": {
          "machineType": "n1-standard-4",
          "acceleratorType": "NVIDIA_TESLA_V100",
          "acceleratorCount": 2
        },
        "pythonPackageSpec": {
          "executorImageUri": "us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest",
          "packageUris": GCS_PATH_FOR_PYTHON_CODE,
          "pythonModule": "trainer.task",
          "args": ["--epochs=50", "--dropout_rate=0.45"]
        }
      }
    ],
    "baseOutputDirectory": {
      "outputUriPrefix": GCS_PATH_FOR_TENSORBOARD_LOG
    },
    "serviceAccount": SERVICE_ACCOUNT_FROM_THE_PERMISSION_STEP,
    "tensorboard": FULL_RESOURCE_NAME_OF_TENSORBOARD
  }
}