Используйте оптимизатор pytorch, чтобы соответствовать пользовательской функции - PullRequest
0 голосов
/ 13 января 2020

Я прочитал много уроков о том, как использовать PyTorch для регрессии по набору данных, используя, например, модель, состоящую из нескольких линейных слоев и потери MSE.

Хорошо, представьте, что я знать, что функция F зависит от переменной x и некоторых неизвестных параметров (p_j: j = 0, ..., P-1), причем P относительно мало, но функция представляет собой состав специальной функции. Итак, моя проблема заключается в классической минимизации, зная данные {x_i, y_i} _i <= N </p>

Min_{ {p_j} } Sum_i (F(x_i;{p_j}) - y_i)^2

Я хотел бы знать, могу ли я использовать оптимизаторы PyTorch, и если да, то как я могу это сделать?

Спасибо.

1 Ответ

0 голосов
/ 17 января 2020

Фактически, эксперты PyTorch отвечают, что минимизируемая функция должна быть выражена через torch.tensors, чтобы минимизаторы могли вычислять градиенты. Так что в моем случае это невозможно.

...