mnist数据识别-正则化 03

qq_41005604

于 2021-03-31 15:09:20 发布

阅读量357

点赞数

分类专栏： Pytorch CNN LSTM

本文链接：https://blog.csdn.net/qq_41005604/article/details/115351036

版权

Pytorch CNN LSTM 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

正则化主要是在定义优化器的时候使用

改变的地方放在这里

LR = 0.5
# 定义模型
model = Net()
# 定义代价函数
mse_loss = nn.CrossEntropyLoss()
# 定义优化器,设置L2正则化
'''
只需要改这个地方就可以了
'''
optimizer = optim.SGD(model.parameters(), LR, weight_decay=0.001)

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

qq_41005604

关注关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
mnist数据识别-正则化 03

正则化主要是在定义优化器的时候使用改变的地方放在这里LR = 0.5# 定义模型model = Net()# 定义代价函数mse_loss = nn.CrossEntropyLoss()# 定义优化器,设置L2正则化'''只需要改这个地方就可以了'''optimizer = optim.SGD(model.parameters(), LR, weight_decay=0.001)...
复制链接

扫一扫