keras.utils.plot_model(simple_model, 'flower_model_with_shape_info.png', show_shapes=True)
simple_model是定义的网络模型
flower_model_with_shape_info.png为保存网络模型的结构图
keras.utils.plot_model(simple_model, 'flower_model_with_shape_info.png', show_shapes=True)
simple_model是定义的网络模型
flower_model_with_shape_info.png为保存网络模型的结构图