keras-team/keras

`plot_model` does not work for all models in `keras.applications`

Star9daisy opened this issue · 4 comments

Hi developers, I follow the tutorial of transfer learning here. After that, I use keras.utils.plot_model to show its structure. However, I find that it doesn't work for all provided models.

Here's a colab notebook I create to show each case and the error message: https://colab.research.google.com/drive/1nTYEHGSXWixq6Bx_ubtgP12NOdXdVWzN?usp=sharing

I'm able to plot the model, you can set the dpi argument in plot_model to smaller number to fit into the frame.

Attached Gist here for reference https://colab.sandbox.google.com/gist/sachinprasadhs/79afe3e25aec48fa350700955ead74f0/transfer_learning.ipynb