首页 > 其他 > 详细

nn.MSELoss() 无输出?原因是忘记初始化!

时间:2020-05-24 17:55:21      阅读:247      评论:0      收藏:0      [点我收藏+]

nn.MSELoss()

昨天用mse实现一个简单的loss时,输出竟然是mse()!无报错
当时真的懵逼了,print输入是有值有梯度的,折腾了大半天,最终是在同学的帮助下解决了这个问题。

当输出是MSE()时,不要怀疑,就是没有初始化!

官方的定义

torch.nn.MSELoss(size_average=None, reduce=None, reduction=‘mean‘)

size_average Default:True

Default: 平均minibatch中的每个loss,如果设为False,则是相加。

reduce Default:True

对于每个minibatch,根据size_average对loss进行平均或求和。

reduction: Default:‘mean‘

‘none‘|‘mean‘|‘sum‘ 应用于输出的状态

实际使用中,一定要注意设置两个False!

loss_fn = nn.MSELoss(reduce=False, size_average=False)

nn.MSELoss() 无输出?原因是忘记初始化!

原文:https://www.cnblogs.com/Wiikk/p/12951617.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!