【Pytorch】查看模型结构、参数和各种属性

【Pytorch】查看模型结构、参数和各种属性

• 一、torchsummary模块
• 二、torchstat模块
• 三、model.children()和model.modules()
• 四、model.parameters()和model.named_parameters()
• 五、model.state_dict()
• 六、m.__class__.__name__

一、torchsummary模块 二、torchstat模块 三、model.children()和model.modules() 四、model.parameters()和model.named_parameters() 五、model.state_dict() 六、m.class.name

获取module的classname。

import torch
conv = torch.nn.Conv2d(1,8,3,1)
bn = torch.nn.BatchNorm2d(8)
l = [conv,bn]
for item in l:
    print(item.__class__.__name__)
# 输出结果：
# Conv2d
# BatchNorm2d