tf.contrib.framework.VariableDeviceChooser.__init__()

tf.contrib.framework.VariableDeviceChooser.__init__(num_tasks=0, job_name='ps', device_type='CPU', device_index=0)

Initialize VariableDeviceChooser.

Usage:

To use with 2 parameter servers: VariableDeviceChooser(2)

To use without parameter servers: VariableDeviceChooser() VariableDeviceChooser(device_type='GPU') # For GPU placement

Args:
  • num_tasks: number of tasks.
  • job_name: String, a name for the parameter server job.
  • device_type: Optional device type string (e.g. "CPU" or "GPU")
  • device_index: int. Optional device index. If left unspecified, device represents 'any' device_index.
doc_TensorFlow
2016-10-14 13:04:50
Comments
Leave a Comment

Please login to continue.