fix: Add newest model CLOUD_1 to avoid customers getting CUDA errors …

…with … (#3408) * Add newest model CLOUD_1 to avoid customers getting CUDA errors with older models. * fix: Add newest model CLOUD_1 to avoid customers getting CUDA errors with older models. --------- Co-authored-by: Karl Weinmeister <11586922+kweinmeister@users.noreply.github.com> Co-authored-by: Patti Shin <pattishin@users.noreply.github.com>
GoogleCloudPlatform · Sep 11, 2023 · dba9baa · dba9baa
1 parent fc2d0b4
commit dba9baa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ai-platform/snippets/create-training-pipeline-image-object-detection.js b/ai-platform/snippets/create-training-pipeline-image-object-detection.js
@@ -58,7 +58,7 @@ async function main(
     const trainingTaskInputsObj =
       new definition.AutoMlImageObjectDetectionInputs({
         disableEarlyStopping: false,
-        modelType: ModelType.CLOUD_HIGH_ACCURACY_1,
+        modelType: ModelType.CLOUD_1,
         budgetMilliNodeHours: 20000,
       });