获取模型对象1
res = model.decoder
获取模型参数对象2
res = model.decoder.parameters()
提示:
我们可以通过 Module 类的 parameters() 或者 named_parameters ⽅法来访问所有参数(以迭代器的形式返回),后者除了返回参数 Tensor 外还会返回其名字。
打印参数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]
本文介绍了如何在PyTorch中管理和访问模型参数,包括使用parameters()和named_parameters()方法来迭代访问模型的所有参数,以及如何通过state_dict()保存和加载模型参数。此外还讲解了如何锁定特定层的参数以防止其在训练过程中被更新。
3万+

被折叠的 条评论
为什么被折叠?



