From 581939b1879ca6b47af482883fe0dfbec089f804 Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Fri, 24 Feb 2023 11:04:25 -0800 Subject: [PATCH] feat: Add the recently added new model type "cloud_1" to the "AutoMLImageTrainingJob" in SDK. PiperOrigin-RevId: 512114110 --- google/cloud/aiplatform/constants/base.py | 4 ++-- google/cloud/aiplatform/training_jobs.py | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/google/cloud/aiplatform/constants/base.py b/google/cloud/aiplatform/constants/base.py index 1bdf919615..c8af42c383 100644 --- a/google/cloud/aiplatform/constants/base.py +++ b/google/cloud/aiplatform/constants/base.py @@ -81,8 +81,8 @@ # NOTE: When adding a new prediction_type's, ensure it fits the pattern # "automl_image_{prediction_type}_*" used by the YAML schemas on GCS AUTOML_IMAGE_PREDICTION_MODEL_TYPES = { - "classification": {"CLOUD"} | MOBILE_TF_MODEL_TYPES, - "object_detection": {"CLOUD_HIGH_ACCURACY_1", "CLOUD_LOW_LATENCY_1"} + "classification": {"CLOUD", "CLOUD_1"} | MOBILE_TF_MODEL_TYPES, + "object_detection": {"CLOUD_1", "CLOUD_HIGH_ACCURACY_1", "CLOUD_LOW_LATENCY_1"} | MOBILE_TF_MODEL_TYPES, } diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py index 3b3db8ff6c..83880c3fa8 100644 --- a/google/cloud/aiplatform/training_jobs.py +++ b/google/cloud/aiplatform/training_jobs.py @@ -5276,6 +5276,9 @@ def __init__( "CLOUD" - Default for Image Classification. A Model best tailored to be used within Google Cloud, and which cannot be exported. + "CLOUD_1" - A model type best tailored to be used within Google Cloud, + which cannot be exported externally. Compared to the CLOUD model + above, it is expected to have higher prediction accuracy. "CLOUD_HIGH_ACCURACY_1" - Default for Image Object Detection. A model best tailored to be used within Google Cloud, and which cannot be exported. Expected to have a higher latency,