PyTorchConfiguration Class
Represents configuration information for distributed PyTorch jobs.
This class is used in the RunConfiguration class.
Class PyTorchConfiguration constructor.
- Inheritance
-
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElementPyTorchConfiguration
Constructor
PyTorchConfiguration(communication_backend='Nccl', process_count=None, node_count=1)
Parameters
Name | Description |
---|---|
communication_backend
|
Communication backend to use for PyTorch distributed training. Supported backends are "Nccl" and "Gloo". Default is "Nccl". default value: Nccl
|
process_count
|
The total number of processes to launch for the job. By default the
value will be set to the default value: None
|
node_count
|
The number of nodes to use for the job. default value: 1
|
الملاحظات
https://aka.ms/ContentUserFeedback.
قريبًا: خلال عام 2024، سنتخلص تدريجيًا من GitHub Issues بوصفها آلية إرسال ملاحظات للمحتوى ونستبدلها بنظام ملاحظات جديد. لمزيد من المعلومات، راجعإرسال الملاحظات وعرضها المتعلقة بـ