损失函数与反向传播

MESLoss

• 导入

from torch import nn

• 用法

# 以均方误差MSE为例
loss_mse = nn.MSELoss()
loss = loss_mse(result, target)

反向传播

• 用法

loss.backward()

补充

要根据实际输出，选择合适的损失函数。

标签：loss,函数,nn,传播,Pytorch,反向,mse
来源： https://www.cnblogs.com/FreshmanXu/p/15864108.html