获取模型对象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]