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
自提交3599fd4起, model_to_dot()
不起作用。 较小的修复似乎可行。
最有用的评论
自提交3599fd4起,
model_to_dot()
不起作用。 较小的修复似乎可行。