Class PyTorchSettings
java.lang.Object
com.microsoft.azure.management.batchai.PyTorchSettings
pyTorch job settings.
-
Constructor Summary
-
Method Summary
Modifier and TypeMethodDescriptionGet command line arguments that need to be passed to the python script.Get type of the communication backend for distributed jobs.Get number of processes to launch for the job execution.Get the path to the Python interpreter.Get the python script to execute.withCommandLineArgs
(String commandLineArgs) Set command line arguments that need to be passed to the python script.withCommunicationBackend
(String communicationBackend) Set type of the communication backend for distributed jobs.withProcessCount
(Integer processCount) Set number of processes to launch for the job execution.withPythonInterpreterPath
(String pythonInterpreterPath) Set the path to the Python interpreter.withPythonScriptFilePath
(String pythonScriptFilePath) Set the python script to execute.
-
Constructor Details
-
PyTorchSettings
public PyTorchSettings()
-
-
Method Details
-
pythonScriptFilePath
Get the python script to execute.- Returns:
- the pythonScriptFilePath value
-
withPythonScriptFilePath
Set the python script to execute.- Parameters:
pythonScriptFilePath
- the pythonScriptFilePath value to set- Returns:
- the PyTorchSettings object itself.
-
pythonInterpreterPath
Get the path to the Python interpreter.- Returns:
- the pythonInterpreterPath value
-
withPythonInterpreterPath
Set the path to the Python interpreter.- Parameters:
pythonInterpreterPath
- the pythonInterpreterPath value to set- Returns:
- the PyTorchSettings object itself.
-
commandLineArgs
Get command line arguments that need to be passed to the python script.- Returns:
- the commandLineArgs value
-
withCommandLineArgs
Set command line arguments that need to be passed to the python script.- Parameters:
commandLineArgs
- the commandLineArgs value to set- Returns:
- the PyTorchSettings object itself.
-
processCount
Get number of processes to launch for the job execution. The default value for this property is equal to nodeCount property.- Returns:
- the processCount value
-
withProcessCount
Set number of processes to launch for the job execution. The default value for this property is equal to nodeCount property.- Parameters:
processCount
- the processCount value to set- Returns:
- the PyTorchSettings object itself.
-
communicationBackend
Get type of the communication backend for distributed jobs. Valid values are 'TCP', 'Gloo' or 'MPI'. Not required for non-distributed jobs.- Returns:
- the communicationBackend value
-
withCommunicationBackend
Set type of the communication backend for distributed jobs. Valid values are 'TCP', 'Gloo' or 'MPI'. Not required for non-distributed jobs.- Parameters:
communicationBackend
- the communicationBackend value to set- Returns:
- the PyTorchSettings object itself.
-