From e089f3fd4c1087c2363d620f53ab29c18b7909a8 Mon Sep 17 00:00:00 2001 From: Takashi Miyauchi Date: Fri, 14 Jul 2023 11:42:44 +0900 Subject: [PATCH] Modify instance_count and instance_type to use Pipelines's Parameter in batch-inference/seedcode/pipelines/batch_inference/pipeline.py --- .../seedcode/pipelines/batch_inference/pipeline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/batch-inference/seedcode/pipelines/batch_inference/pipeline.py b/batch-inference/seedcode/pipelines/batch_inference/pipeline.py index 822912b9..5c62a8eb 100644 --- a/batch-inference/seedcode/pipelines/batch_inference/pipeline.py +++ b/batch-inference/seedcode/pipelines/batch_inference/pipeline.py @@ -92,8 +92,8 @@ def get_pipeline( #### SAGEMAKER CONSTRUCTS transform = Transformer( model_name=model_name, - instance_count=1, - instance_type='ml.m5.xlarge', + instance_count=batch_inference_instance_count, + instance_type=batch_inference_instance_type, output_path=output_path, base_transform_job_name=f"{base_job_prefix}/batch-transform-job", max_payload=10,