PyTorchDistribution Klass
PyTorch-distributionskonfiguration.
- Arv
-
azure.ai.ml.entities._job.distribution.DistributionConfigurationPyTorchDistribution
Konstruktor
PyTorchDistribution(*, process_count_per_instance: int | None = None, **kwargs: Any)
Keyword-Only parametrar
Name | Description |
---|---|
process_count_per_instance
|
Antalet processer per nod. |
Exempel
Konfigurera en CommandComponent med en 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,
)
Variabler
Name | Description |
---|---|
type
|
Anger typ av distribution. Ange automatiskt till "pytorch" för den här klassen. |
Samarbeta med oss på GitHub
Källan för det här innehållet finns på GitHub, där du även kan skapa och granska ärenden och pull-begäranden. Se vår deltagarguide för mer information.
Azure SDK for Python
Feedback
https://aka.ms/ContentUserFeedback.
Kommer snart: Under hela 2024 kommer vi att fasa ut GitHub-problem som feedbackmekanism för innehåll och ersätta det med ett nytt feedbacksystem. Mer information finns i:Skicka och visa feedback för