tf.keras.utils.plot_model(model, 'my_model.png', show_shapes=True)