PyTorchSettings Class

pyTorch job settings.

All required parameters must be populated in order to send to Azure.

Inheritance

Model

PyTorchSettings

Constructor

PyTorchSettings(*, python_script_file_path: str, python_interpreter_path: Optional[str] = None, command_line_args: Optional[str] = None, process_count: Optional[int] = None, communication_backend: Optional[str] = None, **kwargs)

Parameters

python_script_file_path

str

Required

Required. The python script to execute.

python_interpreter_path

str

Required

The path to the Python interpreter.

command_line_args

str

Required

Command line arguments that need to be passed to the python script.

process_count

int

Required

Number of processes to launch for the job execution. The default value for this property is equal to nodeCount property.

communication_backend

str

Required

Type of the communication backend for distributed jobs. Valid values are 'TCP', 'Gloo' or 'MPI'. Not required for non-distributed jobs.