Вычислить функцию внешних потерь, но вычислить градиенты в тензорном потоке? - PullRequest
0 голосов
/ 24 августа 2018

Я хочу обучить модель, для которой функция потерь может быть вычислена только внешне.Итак, я беру выходные данные моего последнего слоя, вычисляю внешнее значение и хочу использовать это для обновления моей сети.Могу ли я реализовать такую ​​настройку в tenorflow?

1 Ответ

0 голосов
/ 24 августа 2018

Да, вы можете, вы должны определить минимальные потери из графика, например:

loss = Network.loss(input_tensor)
loss_2 = out_function(loss)

И тогда вам просто нужно указать:

train_step = optimizer.minimize(loss_2)

Должно работать

...