1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
# 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
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
|
