PyTorchDistribution Class
PyTorch distribution configuration.
- Inheritance
-
azure.ai.ml.entities._job.distribution.DistributionConfigurationPyTorchDistribution
Constructor
PyTorchDistribution(*, process_count_per_instance: int | None = None, **kwargs: Any)
Keyword-Only Parameters
Name | Description |
---|---|
process_count_per_instance
|
The number of processes per node. |
Examples
Configuring a CommandComponent with a PyTorchDistribution.
from azure.ai.ml import PyTorchDistribution
from azure.ai.ml.entities import CommandComponent
component = CommandComponent(
name="microsoftsamples_torch",
description="This is the PyTorch command component",
inputs={
"component_in_number": {"description": "A number", "type": "number", "default": 10.99},
"component_in_path": {"description": "A path", "type": "uri_folder"},
},
outputs={"component_out_path": {"type": "uri_folder"}},
command="echo Hello World & echo ${{inputs.component_in_number}} & echo ${{inputs.component_in_path}} "
"& echo ${{outputs.component_out_path}}",
environment="AzureML-sklearn-1.0-ubuntu20.04-py38-cpu:33",
distribution=PyTorchDistribution(
process_count_per_instance=2,
),
instance_count=2,
)
Variables
Name | Description |
---|---|
type
|
Specifies the type of distribution. Set automatically to "pytorch" for this class. |
Collaborate with us on GitHub
The source for this content can be found on GitHub, where you can also create and review issues and pull requests. For more information, see our contributor guide.
Azure SDK for Python
Feedback
https://aka.ms/ContentUserFeedback.
Coming soon: Throughout 2024 we will be phasing out GitHub Issues as the feedback mechanism for content and replacing it with a new feedback system. For more information see:Submit and view feedback for