Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Dot export with Keras
from keras.applications import *
from keras.utils import plot_model
# [..]
# model = ...
# Get your own model here
# [..]
model = NASNetMobile() #Example with NASNetMobile
plot_model(model,show_shapes=False, to_file='model.dot')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment