Skip to content

Can not set instance count when using Batch Endpoint. (From job) #3610

@runtu9527

Description

@runtu9527

Describe your suggestion

https://learn.microsoft.com/en-us/azure/machine-learning/how-to-use-batch-pipeline-from-job?view=azureml-api-2&tabs=cli

I have a Pipeline job with 2 steps. The first step is configured with 2 instance counts. However, when I publish this job to Batch Endpoint, the instance count becomes 1, and there is no other way to set the instance count when calling batch endpoint.

When published to Batch Endpoint, the instance_count and process_count_per_node not works, but the run_invocation_timeout in environment variables is works.

"Jobs": { "flow_step": { "type": "parallel", "resources": { "instance_count": 2 }, "error_threshold": -1, "environment_variables": { "AZUREML_PARAMETER_aml_run_invocation_timeout": "13800", "AZUREML_PARAMETER_aml_run_max_try": "3", "AZUREML_PARAMETER_aml_process_count_per_node": "8", "AZUREML_PARAMETER_aml_max_concurrency_per_instance": "8", "AZUREML_PARAMETER_aml_error_threshold": "-1" }, "max_concurrency_per_instance": 8,

Additional details

My step is a PromptFlow step and is a parallel job step.

Also, in PromptFlow, the instance count settings are after the flow_component.
https://microsoft.github.io/promptflow/cloud/azureai/use-flow-in-azure-ml-pipeline.html#component-ports-and-run-settings

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions