在model中设置了一个可学习的参数,但是在训练的过程中数值一直没能更新,在打印model的parameter检查的时候也没有看到这个参数
self.a = nn.Parameter(torch.ones(5)) #设置可学习参数
for name, p in model.named_parameters(): # 打印检查model的参数print(name)
设置的参数能够被绑定到model中,并且学习、更新数值