‘Model‘ object has no attribute ‘_container_nodes‘

最新推荐文章于 2023-03-10 17:05:43 发布

Dcurry

最新推荐文章于 2023-03-10 17:05:43 发布

阅读量499

点赞数

文章标签： python 深度学习可视化神经网络 tensorflow

本文链接：https://blog.csdn.net/dcurry/article/details/109560323

版权

解决方案：

如果使用的是keras ，
请保留
if node_key in model._container_nodes:

如何是从tf.keras.leyers 导出的：
那么按如下修改

Fix to tensorflow#17633***（https://github.com/tensorflow/tensorflow/issues/17633）***

‘Model’ object has no attribute ‘_container_nodes’ error when using tf.keras.utils.plot_model().
Replaced
if node_key in model._container_nodes:
with
if node_key in model._network_nodes: # pylint: disable=protected-access
in tensorflow\python\keras_impl\keras\utils\vis_utils.py.