孤独小蝴蝶

文章
2
资源
0
加入时间
2年10月17天

Pytorch的add_param_group使用说明

这个可以用来冻结某些参数,也可以用来指定参数的学习import torchimport torch.optim as optimw1 = torch.randn(3, 3)w1.requires_grad = Truew2 = torch.randn(3, 3)w2.requires_grad = Trueo = optim.Adam([w1])print(o.param_gr...