com.microsoft.azure.management.batchai

Class TensorFlowSettings

java.lang.Object

com.microsoft.azure.management.batchai.TensorFlowSettings

public class TensorFlowSettings
extends Object

TensorFlow job settings.

Constructor Summary

Constructors

Constructor and Description
TensorFlowSettings()

Method Summary

Modifier and Type	Method and Description
String	masterCommandLineArgs() Get command line arguments that need to be passed to the python script for the master task.
String	parameterServerCommandLineArgs() Get command line arguments that need to be passed to the python script for the parameter server.
Integer	parameterServerCount() Get the number of parameter server tasks.
String	pythonInterpreterPath() Get the path to the Python interpreter.
String	pythonScriptFilePath() Get the python script to execute.
TensorFlowSettings	withMasterCommandLineArgs(String masterCommandLineArgs) Set command line arguments that need to be passed to the python script for the master task.
TensorFlowSettings	withParameterServerCommandLineArgs(String parameterServerCommandLineArgs) Set command line arguments that need to be passed to the python script for the parameter server.
TensorFlowSettings	withParameterServerCount(Integer parameterServerCount) Set the number of parameter server tasks.
TensorFlowSettings	withPythonInterpreterPath(String pythonInterpreterPath) Set the path to the Python interpreter.
TensorFlowSettings	withPythonScriptFilePath(String pythonScriptFilePath) Set the python script to execute.
TensorFlowSettings	withWorkerCommandLineArgs(String workerCommandLineArgs) Set command line arguments that need to be passed to the python script for the worker task.
TensorFlowSettings	withWorkerCount(Integer workerCount) Set the number of worker tasks.
String	workerCommandLineArgs() Get command line arguments that need to be passed to the python script for the worker task.
Integer	workerCount() Get the number of worker tasks.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TensorFlowSettings

public TensorFlowSettings()

Method Detail

pythonScriptFilePath

public String pythonScriptFilePath()

Get the python script to execute.

Returns:

the pythonScriptFilePath value

withPythonScriptFilePath

public TensorFlowSettings withPythonScriptFilePath(String pythonScriptFilePath)

Set the python script to execute.

Parameters:

pythonScriptFilePath - the pythonScriptFilePath value to set

Returns:

the TensorFlowSettings object itself.

pythonInterpreterPath

public String pythonInterpreterPath()

Get the path to the Python interpreter.

Returns:

the pythonInterpreterPath value

withPythonInterpreterPath

public TensorFlowSettings withPythonInterpreterPath(String pythonInterpreterPath)

Set the path to the Python interpreter.

Parameters:

pythonInterpreterPath - the pythonInterpreterPath value to set

Returns:

the TensorFlowSettings object itself.

masterCommandLineArgs

public String masterCommandLineArgs()

Get command line arguments that need to be passed to the python script for the master task.

Returns:

the masterCommandLineArgs value

withMasterCommandLineArgs

public TensorFlowSettings withMasterCommandLineArgs(String masterCommandLineArgs)

Set command line arguments that need to be passed to the python script for the master task.

Parameters:

masterCommandLineArgs - the masterCommandLineArgs value to set

Returns:

the TensorFlowSettings object itself.

workerCommandLineArgs

public String workerCommandLineArgs()

Get command line arguments that need to be passed to the python script for the worker task. Optional for single process jobs.

Returns:

the workerCommandLineArgs value

withWorkerCommandLineArgs

public TensorFlowSettings withWorkerCommandLineArgs(String workerCommandLineArgs)

Set command line arguments that need to be passed to the python script for the worker task. Optional for single process jobs.

Parameters:

workerCommandLineArgs - the workerCommandLineArgs value to set

Returns:

the TensorFlowSettings object itself.

parameterServerCommandLineArgs

public String parameterServerCommandLineArgs()

Get command line arguments that need to be passed to the python script for the parameter server. Optional for single process jobs.

Returns:

the parameterServerCommandLineArgs value

withParameterServerCommandLineArgs

public TensorFlowSettings withParameterServerCommandLineArgs(String parameterServerCommandLineArgs)

Set command line arguments that need to be passed to the python script for the parameter server. Optional for single process jobs.

Parameters:

parameterServerCommandLineArgs - the parameterServerCommandLineArgs value to set

Returns:

the TensorFlowSettings object itself.

workerCount

public Integer workerCount()

Get the number of worker tasks. If specified, the value must be less than or equal to (nodeCount * numberOfGPUs per VM). If not specified, the default value is equal to nodeCount. This property can be specified only for distributed TensorFlow training.

Returns:

the workerCount value

withWorkerCount

public TensorFlowSettings withWorkerCount(Integer workerCount)

Set the number of worker tasks. If specified, the value must be less than or equal to (nodeCount * numberOfGPUs per VM). If not specified, the default value is equal to nodeCount. This property can be specified only for distributed TensorFlow training.

Parameters:

workerCount - the workerCount value to set

Returns:

the TensorFlowSettings object itself.

parameterServerCount

public Integer parameterServerCount()

Get the number of parameter server tasks. If specified, the value must be less than or equal to nodeCount. If not specified, the default value is equal to 1 for distributed TensorFlow training. This property can be specified only for distributed TensorFlow training.

Returns:

the parameterServerCount value

withParameterServerCount

public TensorFlowSettings withParameterServerCount(Integer parameterServerCount)

Set the number of parameter server tasks. If specified, the value must be less than or equal to nodeCount. If not specified, the default value is equal to 1 for distributed TensorFlow training. This property can be specified only for distributed TensorFlow training.

Parameters:

parameterServerCount - the parameterServerCount value to set

Returns:

the TensorFlowSettings object itself.