在原始代码中,标记设置为tf.apps.flags.DEFINE_string( 'master','','要使用的TensorFlow master的地址。')。然后我将tf.app.flags更改为tf.flags
最初的FLAGS = tf.app.flags.FLAGS,类似地更改为tf.flags.FLAGS。
,但是在两种情况下都存在tf.constant错误。如何解决? 我觉得这个错误与python版本有关。但无法弄清楚
replica_id = tf.constant(FLAGS.task,dtype = tf.int32,shape =()),
答案 0 :(得分:1)
尝试一下,对我来说效果很好:
import tensorflow as tf
FLAGS = tf.flags.FLAGS
tf.flags.DEFINE_integer('task', 10, "my value for the constant")
# now define your constant
replica_id = tf.constant(value=FLAGS.task, dtype=tf.float32)
# see if it works:
with tf.Session() as sess:
print(sess.run(replica_id))