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.
Please login to continue.