今天在使用keras进行可视化网络结构的时候出现了下面的错误:

 'InputLayer' object is not iterable

解决方法

发现自己使用的是tf keras不是keras,所以切换成tfkeras的plot_model就行了:

from tensorflow.keras.utils import  plot_model

参考文献

[1].TypeError: ‘InputLayer’ object is not iterable. https://stackoverflow.com/questions/60419040/typeerror-inputlayer-object-is-not-iterable