PyTorchDistribution クラス

PyTorch ディストリビューションの構成。

継承

azure.ai.ml.entities._job.distribution.DistributionConfiguration

PyTorchDistribution

コンストラクター

PyTorchDistribution(*, process_count_per_instance: int | None = None, **kwargs: Any)

キーワードのみのパラメーター

名前	説明
process_count_per_instance	Optional[int] ノードあたりのプロセスの数。

例

PyTorchDistribution を使用した CommandComponent の構成。

   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,
   )

変数

名前	説明
type	str 分布の種類を指定します。 このクラスの "pytorch" に自動的に設定します。