def cal_loss(self, pred, target):
#返回加了正则的损失
''' Calculate loss '''
regularization_loss = 0
for param in model.parameters():
# TODO: you may implement L1/L2 regularization here
# 使用L2正则项
# regularization_loss += torch.sum(abs(param))
regularization_loss += torch.sum(param ** 2)
return self.criterion(pred, target) + 0.00075 * regularization_loss
对损失函数加入正则
最新推荐文章于 2023-12-31 01:38:36 发布