if loss_function == 'MSE': loss_func = nn.MSELoss(reduction='mean')，怎么改成MAE的代码

如果你想在 PyTorch 中使用 MAE 作为损失函数，可以使用 `nn.L1Loss()`。`nn.L1Loss()` 计算的是预测值与真实值之间的绝对误差，也称为平均绝对误差（MAE）。

以下是将 MSE 损失函数改为 MAE 损失函数的代码示例：

import torch.nn as nn

if loss_function == 'MSE':
    loss_func = nn.MSELoss(reduction='mean')
elif loss_function == 'MAE':
    loss_func = nn.L1Loss(reduction='mean')

可以看到，只需要将 `nn.MSELoss()` 替换为 `nn.L1Loss()` 即可。其他的参数设置和使用方法与 MSE 损失函数相同。