如果使用的是keras , 请保留 if node_key in model._container_nodes:
如何是从tf.keras.leyers 导出的: 那么按如下修改
Fix to tensorflow#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.
如果使用的是keras ,请保留if node_key in model._container_nodes:如何是从tf.keras.leyers 导出的:那么按如下修改Fix to tensorflow#17633'Model' object has no attribute '_container_nodes' error when using tf.keras.utils.plot...