Я реализую некоторый RL в PyTorch, и мне пришлось написать свою собственную функцию mse_loss (которую я нашел в Stackoverflow;)). Функция потерь:
def mse_loss(input_, target_):
return torch.sum(
(input_ - target_) * (input_ - target_)) / input_.data.nelement()
Теперь, в моем тренировочном цикле, первый ввод выглядит примерно так:
tensor([-1.7610e+10]), tensor([-6.5097e+10])
С этим вводом я получу ошибку:
Unable to get repr for <class 'torch.Tensor'>
Вычисление a = (input_ - target_)
работает нормально, а b = a * a
соответственно b = torch.pow(a, 2)
завершится ошибкой, указанной выше.
Кто-нибудь знает, как это исправить?
Большое спасибо!
Обновление: я только что попытался использовать torch.nn.functional.mse_loss
, что приведет к той же ошибке..