[pytorch]网络结构查询和提取

最新推荐文章于 2023-06-21 15:41:35 发布

一骑红尘荔枝来

最新推荐文章于 2023-06-21 15:41:35 发布

阅读量236

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/condom10010/article/details/129535654

版权

Pytorch提取神经网络层结构、层参数及自定义初始化

resnet中一个block的参数

结构:
在这里插入图片描述

参数:查询结果

BasicBlock(
  (conv1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
  (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
  (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
  (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
  (shortcut): Sequential()
)