pytorch中Module模块中named_parameters函数

最新推荐文章于 2024-07-25 22:32:26 发布

小白827

最新推荐文章于 2024-07-25 22:32:26 发布

阅读量6.6k

点赞数 2

分类专栏： pytorch

本文链接：https://blog.csdn.net/qq_24503095/article/details/103630382

版权

本文介绍PyTorch中Module模块的named_parameters()函数，该函数用于获取模型中所有参数的名称及其对应的数值，对于理解和调试深度学习模型的参数非常有用。

摘要由CSDN通过智能技术生成

函数named_parameters()，返回各层中参数名称和数据

class MLP(nn.Module):
    def __init__(self):
        super(MLP, self).__init__()
        self.hidden = nn.Sequential(
            nn.Linear(256,64),
            nn.ReLU(inplace=True),
            nn.Linear(64,10)
                )
    def forward(self, x):
        return self.hidden(x)

    
#x = torch.randperm(256*2).view(-1,256)
net = MLP()
net(x