PyTorchConfiguration クラス

分散 PyTorch ジョブの構成情報を表します。

このクラスは、 RunConfiguration クラスで使用されます。

クラス PyTorchConfiguration コンストラクター。

コンストラクター

PyTorchConfiguration(communication_backend='Nccl', process_count=None, node_count=1)

名前	説明
communication_backend	str PyTorch 分散トレーニングに使用する通信バックエンド。サポートされているバックエンドは、"Nccl" と "Gloo" です。既定値は "Nccl" です。規定値: Nccl
process_count	int ジョブの起動するプロセスの合計数。既定では、値は `node_count`に設定されます。規定値: None
node_count	int ジョブに使用するノードの数。規定値: 1