使用plot_model tensorflow.keras可视化模型

Question

>>> from tensorflow.keras.applications import VGG16
>>> from tensorflow.keras.utils import plot_model
>>> plot_model(VGG16, to_file="vgg16.png", show_shapes=True)

AttributeError: 'function' object has no attribute '_is_graph_network'

我尝试更改show_shapes = False，但是出现相同的错误。我能够为定制模型运行程序。