Tensorflow2打印模型结构图

from tensorflow.keras.utils import plot_model
plot_model(model, to_file='model.png', show_shapes=True)  # 保存模型结构图

 

你可能感兴趣的:(深度学习,算法)