Keras之查看模型layer的层数

最新推荐文章于 2024-02-08 09:59:19 发布

Regnaiq

最新推荐文章于 2024-02-08 09:59:19 发布

阅读量1.3k

点赞数 2

分类专栏： Deep learning

本文链接：https://blog.csdn.net/yq_forever/article/details/119043471

版权

Deep learning 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

当模型很大的时候有时候需要查看模型某个layer对应的层数. 用model.summary()不大好使.

可以用如下代码列举模型的层数:

{i: v for i, v in enumerate(model.layers)}

{0: <keras.layers.convolutional.Conv2D at 0x7f182bcd1710>,
 1: <keras.layers.normalization.BatchNormalization at 0x7f1840981828>,
 2: <keras.layers.advanced_activations.LeakyReLU at 0x7f184154b5c0>,
 3: <keras.layers.pooling.MaxPooling2D at 0x7f184154be10>,
 4: <keras.layers.core.Dropout at 0x7f184154be80>,
 5: <keras.layers.convolutional.Conv2D at 0x7f18800593c8>}