损失函数与反向传播

MESLoss

• 导入

from torch import nn

• 用法

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

反向传播

• 用法

loss.backward()

补充

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