We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ae27422 commit 9fa300eCopy full SHA for 9fa300e
ADAG/ADAG.py
@@ -50,8 +50,7 @@ def main():
50
local_step = tf.Variable(0,dtype=tf.int32,trainable=False,
51
name='local_step',collections=['local_non_trainable'])
52
53
- with tf.device(tf.train.replica_device_setter(
54
- ps_tasks=n_pss,
+ with tf.device(tf.train.replica_device_setter(ps_tasks=n_pss,
55
worker_device="/job:%s/task:%d" % (FLAGS.job_name,FLAGS.task_index))):
56
global_step = tf.Variable(0,dtype=tf.int32,trainable=False,name='global_step')
57
target = tf.constant(100.,shape=[2],dtype=tf.float32)
0 commit comments