Потеря реконструкции Pytorch

Question

Если у меня есть два тензора

truth = [N, 1, 224, 224]

net_output = [N, 1, 224, 224]

и я хочу минимизировать net_output, чтобы соответствовать истине, это правильный способ применить потерю?

mse = nn.L1Loss().cuda()
FPL = mse(truth, net_output.detach())

N размер партии.Будет ли моя потеря l1 суммироваться по N изображениям?

Answer 1

Вы можете использовать потери L1.Я бы сделал это так:

import torch.nn as nn

l1_loss = nn.L1Loss()
loss = mse_loss(input, target)

Вы должны изменить аргументы truth и net_output.detach().