Package 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 Details

    • TensorFlowSettings

      public TensorFlowSettings()

  • Method Details

    • 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.