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