如何修复 tf.constant 意外参数错误

Question

在原始代码中，标志设置为 tf.apps.flags.DEFINE_string( 'master', '', 'The address of the TensorFlow master to use.').然后我将 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=()),

Answer 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))

如何修复 tf.constant 意外参数错误

how to fix tf.constant unexpected argument error

tensorflow

resnet