python pyrotch模型中的参数访问

获取模型对象1

res = model.decoder

获取模型参数对象2

res = model.decoder.parameters()

提示:
我们可以通过 Module 类的 parameters() 或者 named_parameters ⽅法来访问所有参数(以迭代器的形式返回),后者除了返回参数 Tensor 外还会返回其名字。

Pytorch中state_dict()、named_parameters()和parameters()的区别:optim.step只能更新nn.parameter.Parameter类型的参数。不可学习参数将会通过Module.register_parameter()注册在self._buffers中。named_parameters 保存了参数名与具体值,因此可用于锁住某些层的参数,让其在训练的时候不更新参数

打印参数3

print(type(model.named_parameters()))
for name, param in model.named_parameters():
    print(name, param.size())

此时能在Debug4中看到param的值,也可获取模型的第i层,并查看其中的参数:

res = MLP.model
weight_of_i = list( res[i].parameters())[0]
bias_of_i = list( res[i].parameters())[1]

打印模型中的参数个数


  1. 在这里插入图片描述 ↩︎

  2. 在这里插入图片描述 ↩︎

  3. 在这里插入图片描述 ↩︎

  4. 在这里插入图片描述 ↩︎

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值