OS平台和发行版(例如,Linux Ubuntu 16.04):Linux Ubuntu 16.04.1
从(源代码或二进制代码)安装的TensorFlow:二进制文件
TensorFlow版本(使用下面的命令):tensorflow-gpu(1.1.0)
Python版本:2.7.12
我使用此命令来冻结我的图形和权利:
python -m tensorflow.python.tools.freeze_graph \
--input_graph ./rnn_graph.pb \
--input_checkpoint ./model.ckpt-320000 \
--output_graph ./graph_frozen.pb \
--output_node_name=rnn_model_20/softmax_output \
--input_binary=true
我在另一个程序中使用了graph_frozen.pb。我得到了正确的结果。但是当我使用tfdbg和invoke_stepper模型时,我得到了一个非常奇怪的错误:
InvalidArgumentError: A cross-device loop must have a pivot predicate
为什么会出现错误,但我能得到正确答案?我在互联网上找不到同样的问题。有没有人对如何解决这个问题有任何建议? 提前谢谢。