как исправить непредвиденную ошибку аргумента tf.constant - PullRequest
Новая
       7

как исправить непредвиденную ошибку аргумента tf.constant

0 голосов
/ 22 января 2019

В исходном коде флаги были установлены как tf.apps.flags.DEFINE_string ( 'master', '', 'Адрес используемого мастера TensorFlow.'). затем я изменил 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 голосов
/ 22 января 2019

Попробуйте, у меня все работает просто отлично: 

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