Tensorflow: “模型”对象没有属性“ container_nodes”

创建于 2017-11-14  ·  1评论  ·  资料来源: tensorflow/tensorflow

问题

model = tf.keras.models.Model()
model.add(...)
tf.keras.utils.plot_model(model, to_file="model.png")

输出:

Traceback (most recent call last):
  File "model.py", line 36, in <module>
    K.utils.plot_model(model, to_file="model.png")
  File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/keras/_impl/keras/utils/vis_utils.py", line 148, in plot_model
    dot = model_to_dot(model, show_shapes, show_layer_names, rankdir)
  File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/keras/_impl/keras/utils/vis_utils.py", line 123, in model_to_dot
    if node_key in model.container_nodes:
AttributeError: 'Model' object has no attribute 'container_nodes'

环境

-系统:Ubuntu 16.04
-Tensorflow-gpu bin v1.4.0-rc1-11-g130a514 1.4.0

awaiting tensorflower bug

最有用的评论

自提交3599fd4起, model_to_dot()不起作用。 较小的修复似乎可行。

>所有评论

自提交3599fd4起, model_to_dot()不起作用。 较小的修复似乎可行。

此页面是否有帮助?
0 / 5 - 0 等级