PyTorch：模型层

-柚子皮-

模型层的命名方式

命名方式 {自动识别名称如bn_layers}.0.weight

if self.use_bn:     self.bn_layers = nn.ModuleList(         [nn.BatchNorm1d(hidden_units[i + 1]) for i in range(len(hidden_units) - 1)])

 

查看模型每层输出详情

Keras有一个简洁的API来查看模型的每一层输出尺寸，这在调试网络时非常有用。

在PyTorch中也可以实现这个功能。 使用很简单，如下用法： from torchsummary import summary summary(your_model, input_size=(channels, H, W)) input_size 是根据你自己的网络模型的输入尺寸进行设置。

from: -柚子皮-

ref: