arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-14 12:14:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use server_image instead of repeating gcr.io URI

Browse Source
This commit is contained in:
Aurélien Geron
2022-04-17 22:25:53 +12:00
parent 410367eed1
commit e611c9bcf0
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
19_training_and_deploying_at_scale.ipynb
View File

@@ -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",
")"