pytorch手动加入正则
1.Pytorch自带的加正则化方法
Pytorch的优化器都自带正则化,比如
optimizer = optim.Adam(model.parameters(),lr=learning_rate,weight_decay=0.01)
optimizer = optim.SGD(model.parameters(), lr=learing_rate, momentum=0.9,weight_decay=0.01)
其中的weight_decay就是正则化系数,设定为0或者不设定就是无正则化。但这个优化
原创
2021-06-16 17:08:49 ·
4937 阅读 ·
3 评论