Source code for azure.mgmt.batchai.models.py_torch_settings_py3

# coding=utf-8
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
#
# Code generated by Microsoft (R) AutoRest Code Generator.
# Changes may cause incorrect behavior and will be lost if the code is
# regenerated.
# --------------------------------------------------------------------------

from msrest.serialization import Model


[docs]class PyTorchSettings(Model): """Specifies the settings for pyTorch job. All required parameters must be populated in order to send to Azure. :param python_script_file_path: Required. The path and file name of the python script to execute the job. :type python_script_file_path: str :param python_interpreter_path: The path to python interpreter. :type python_interpreter_path: str :param command_line_args: Specifies the command line arguments for the master task. :type command_line_args: str :param process_count: Number of processes to launch for the job execution. The default value for this property is equal to nodeCount property. :type process_count: int :param communication_backend: Type of the communication backend for distributed jobs. Valid values are 'TCP', 'Gloo' or 'MPI'. Not required for non-distributed jobs. :type communication_backend: str """ _validation = { 'python_script_file_path': {'required': True}, } _attribute_map = { 'python_script_file_path': {'key': 'pythonScriptFilePath', 'type': 'str'}, 'python_interpreter_path': {'key': 'pythonInterpreterPath', 'type': 'str'}, 'command_line_args': {'key': 'commandLineArgs', 'type': 'str'}, 'process_count': {'key': 'processCount', 'type': 'int'}, 'communication_backend': {'key': 'communicationBackend', 'type': 'str'}, } def __init__(self, *, python_script_file_path: str, python_interpreter_path: str=None, command_line_args: str=None, process_count: int=None, communication_backend: str=None, **kwargs) -> None: super(PyTorchSettings, self).__init__(**kwargs) self.python_script_file_path = python_script_file_path self.python_interpreter_path = python_interpreter_path self.command_line_args = command_line_args self.process_count = process_count self.communication_backend = communication_backend