From e611c9bcf092ebd658ea2a0ffcaf0d1756ff8b28 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= Date: Sun, 17 Apr 2022 22:25:53 +1200 Subject: [PATCH] Use server_image instead of repeating gcr.io URI --- 19_training_and_deploying_at_scale.ipynb | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/19_training_and_deploying_at_scale.ipynb b/19_training_and_deploying_at_scale.ipynb index 430821e..4205a23 100644 --- a/19_training_and_deploying_at_scale.ipynb +++ b/19_training_and_deploying_at_scale.ipynb @@ -2333,8 +2333,7 @@ " display_name=\"my_custom_training_job\",\n", " script_path=\"my_vertex_ai_training_task.py\",\n", " container_uri=\"gcr.io/cloud-aiplatform/training/tf-gpu.2-4:latest\",\n", - " model_serving_container_image_uri=\n", - " \"gcr.io/cloud-aiplatform/prediction/tf2-gpu.2-8:latest\",\n", + " model_serving_container_image_uri=server_image,\n", " requirements=[\"gcsfs==2022.3.0\"], # not needed, this is just an example\n", " staging_bucket=f\"gs://{bucket_name}/staging\"\n", ")" @@ -2535,8 +2534,7 @@ " display_name=\"my_hp_search_job\",\n", " script_path=\"my_keras_tuner_search.py\",\n", " container_uri=\"gcr.io/cloud-aiplatform/training/tf-gpu.2-4:latest\",\n", - " model_serving_container_image_uri=\n", - " \"gcr.io/cloud-aiplatform/prediction/tf2-gpu.2-8:latest\",\n", + " model_serving_container_image_uri=server_image,\n", " requirements=[\"keras-tuner~=1.1.2\"],\n", " staging_bucket=f\"gs://{bucket_name}/staging\",\n", ")"