在使用均方误差损失函数MSE时
criterion=torch.nn.MSELoss(reduction='mean')
#当reduction=‘none’时,即返回的是一个batch中每个样本的损失
#当reduction='sum’时,即返回一个batch中所有样本损失的和
#当reduction='mean’时,即返回一个batch中所有样本损失的均值
LOSS=criterion(outputs,targets) #标签值
报错
RuntimeError: "mse_cuda" not implemented for 'Long'
这时候需要把outputs和targets的数据类型转化为浮点数float
#这里outputs和targets都是tensor数据类型
outputs=outputs.float()
targets=targets.float()