SageMaker framework processor是否兼容SageMaker Pipelines

0

【以下的问题经过翻译处理】 大家好,

我想问一下,在 Sagemaker pipeline 中是否可以使用 framework processor

我这么问是因为我们必须在调用 .run() 方法时,提交 source_dirframework processor。在将 processor 包装到 Sagemaker.workflow.steps.ProcessingStep 中时,没有可用的参数来指定 source_dir

谢谢! 祝好, Ruoy

profile picture
EXPERT
demandé il y a 8 mois21 vues
1 réponse
0

【以下的回答经过翻译处理】 您可以使用最新版本的sagemaker sdk 2.89.0 来完成这个任务。

from sagemaker.workflow.pipeline_context import PipelineSession

session = PipelineSession()

inputs = [
    ProcessingInput(
    source="s3://my-bucket/sourcefile", 
    destination="/opt/ml/processing/inputs/",),
]

processor = FrameworkProcessor(...)

step_args = processor.run(inputs=inputs, source_dir="...")

step_sklearn = ProcessingStep(
    name="MyProcessingStep",
    step_args=step_args,
)
profile picture
EXPERT
répondu il y a 8 mois

Vous n'êtes pas connecté. Se connecter pour publier une réponse.

Une bonne réponse répond clairement à la question, contient des commentaires constructifs et encourage le développement professionnel de la personne qui pose la question.

Instructions pour répondre aux questions