tensorflow 模型可视化的一点坑

如果使用的是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.