Tensorflow,程序停留在sess.run()函数

时间:2017-09-16 12:17:07

标签: python tensorflow

我是Tensorflow的新手,我遇到了一个问题。当我的程序到达x_batch = sess.run(X_mb)时,它会被卡住(因此,它可以打印1和2,但不能打印4.我猜这是一个无限循环)。我打印X_mb的值,结果为Tensor("batch:0", shape=(32, 39), dtype=float32)。任何人都可以帮助我吗?谢谢! mb_size = 32,g' s形状是[60366,39]。

更新代码:

X = tf.placeholder(tf.float32, shape=[None, X_dim])

def sample_z(m, n):
    return np.random.uniform(-1., 1., size=[m, n])

g = tf.unstack(data, num = 60366, axis = 0)
X_mb, *_ = tf.train.batch(g ,mb_size, capacity = 60366)
sess = tf.train.MonitoredSession()


i = 0

for it in range(2000):
    #print(1)
    for _ in range(5):
        #print(2)       

        #print(X_mb)
        x_batch = sess.run(X_mb)
       # print(4)
        _, D_loss_curr, _ = sess.run(
            [D_solver, D_loss, clip_D],
            feed_dict={X: x_batch, z: sample_z(mb_size, z_dim)}
        )

    _, G_loss_curr = sess.run(
        [G_solver, G_loss],
        feed_dict={z: sample_z(mb_size, z_dim)}
    )

以下是错误消息:

runfile('/Users/franklan123/.spyder-py3/temp.py', wdir='/Users/franklan123/.spyder-py3')
Traceback (most recent call last):

  File "<ipython-input-4-6af95f06eb7f>", line 1, in <module>
    runfile('/Users/franklan123/.spyder-py3/temp.py', wdir='/Users/franklan123/.spyder-py3')

  File "/Users/franklan123/anaconda/envs/tensorflow/lib/python3.6/site-packages/spyder/utils/site/sitecustomize.py", line 688, in runfile
    execfile(filename, namespace)

  File "/Users/franklan123/anaconda/envs/tensorflow/lib/python3.6/site-packages/spyder/utils/site/sitecustomize.py", line 101, in execfile
    exec(compile(f.read(), filename, 'exec'), namespace)

  File "/Users/franklan123/.spyder-py3/temp.py", line 48, in <module>
    X = tf.placeholder(tf.float32, shape=[None, X_dim])

  File "/Users/franklan123/anaconda/envs/tensorflow/lib/python3.6/site-packages/tensorflow/python/ops/array_ops.py", line 1530, in placeholder
    return gen_array_ops._placeholder(dtype=dtype, shape=shape, name=name)

  File "/Users/franklan123/anaconda/envs/tensorflow/lib/python3.6/site-packages/tensorflow/python/ops/gen_array_ops.py", line 1954, in _placeholder
    name=name)

  File "/Users/franklan123/anaconda/envs/tensorflow/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 767, in apply_op
    op_def=op_def)

  File "/Users/franklan123/anaconda/envs/tensorflow/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 2458, in create_op
    self._check_not_finalized()

  File "/Users/franklan123/anaconda/envs/tensorflow/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 2181, in _check_not_finalized
    raise RuntimeError("Graph is finalized and cannot be modified.")

RuntimeError: Graph is finalized and cannot be modified.

1 个答案:

答案 0 :(得分:1)

没有无限循环,tf.train.batch会创建队列,您必须运行队列运行器才能获取任何数据。否则,当你执行sess.run时,你的代码只会挂起queue.dequeue()操作(因为没有线程实际将数据放入队列中)。

简单的解决方案:

  1. 在会话创建之前移动X_mb行(这修改了图表!)
  2. 将tf.Session更改为tf.train.MonitoredSession(将为您运行队列,否则您必须手动运行它们+ MonitoredSession最终确定图表,这样您将避免出现如上所述的错误)。执行此操作后 - 删除具有全局初始化程序的行(MonitoredSession也为您初始化)。