昨天用mse实现一个简单的loss时,输出竟然是mse()!无报错
当时真的懵逼了,print输入是有值有梯度的,折腾了大半天,最终是在同学的帮助下解决了这个问题。
当输出是MSE()时,不要怀疑,就是没有初始化!
size_average Default:True
Default: 平均minibatch中的每个loss,如果设为False,则是相加。
reduce Default:True
对于每个minibatch,根据size_average对loss进行平均或求和。
reduction: Default:‘mean‘
‘none‘|‘mean‘|‘sum‘ 应用于输出的状态
实际使用中,一定要注意设置两个False!
原文:https://www.cnblogs.com/Wiikk/p/12951617.html