From dba9baabfb8b68aeda9c83aa82cd61968a1a0b23 Mon Sep 17 00:00:00 2001 From: Daniel Elias Becerra <37950429+DanElias@users.noreply.github.com> Date: Fri, 18 Aug 2023 18:35:47 +0200 Subject: [PATCH] =?UTF-8?q?fix:=20Add=20newest=20model=20CLOUD=5F1=20to=20?= =?UTF-8?q?avoid=20customers=20getting=20CUDA=20errors=20with=20=E2=80=A6?= =?UTF-8?q?=20(#3408)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * 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 --- .../snippets/create-training-pipeline-image-object-detection.js | 2 +- 1 file changed, 1 insertion(+), 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 index 207fa71915..02f17c36e9 100644 --- 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, });