torch如何查看模型的中间输出结果

最新推荐文章于 2022-10-13 14:27:57 发布

原创

最新推荐文章于 2022-10-13 14:27:57 发布 · 3.8k 阅读

3 ·

CC 4.0 BY-SA版权

在TensorFlow中，我们可以直接通过层的名字查看模型的中间输出，但在PyTorch中，这一操作较为复杂。由于PyTorch无法直接通过名称索引中间层，我们需要遍历模型的层进行访问。本文将介绍如何在PyTorch中查看VGG16模型中某一层，如第一次池化层后的tensor形状和值，以及在复杂网络结构中如何获取特定层的输出。尽管需要通过索引而非名称，但通过`.modules`可以实现这一目标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

博主之前一直是用tensorflow，但是最近有在接触一点torch，发现torch中有一个地方很不方便就是他定义在网络中间的层的结果我们没法看，因为它没有办法像tensorflow这种可以直接通过名字索引。比如：

self.conv1_1 = self.conv_layer(bgr, 3, 64, "conv1_1")
self.conv1_2 = self.conv_layer(self.conv1_1,64,64, "conv1_2")
self.pool1 = self.max_pool(self.conv1_2, 'pool1')

self.conv2_1 = self.conv_layer(self.pool1,64, 128, "conv2_1")
self.conv2_2 = self.conv_layer(self.conv2_1,128, 128, "conv2_2")
self.pool2 = self.max_pool(self.conv2_2, 'pool2')

self.conv3_1 = self.conv_layer(self.pool2,128, 256, "conv3_1")
self.conv3_2 = self.conv_layer(self.conv3_1,256, 256, "conv3_2")
self.conv3_3 = self.conv_layer(self.conv3_2, 256, 256,"conv3_3")
self.pool3 = self.max_pool(self.conv3_3, 'pool3')

这里我们截取了一段Vgg16中的一段代码，我们想看下第一次池化后的我们的tensor的形状那我们直接下面就可以了，