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

时间:2019-01-22 07:59:47

标签: tensorflow resnet

在原始代码中,标记设置为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 =()),

1 个答案:

答案 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))