Torch基準を変更する

Question

TorchでClassNLLCriterionを変更したカスタム損失関数を作成したいとします。具体的には、ClassNLLCriterion損失は次のとおりです。

loss(x, class) = -x[class] 

私はこれがあることを変更したい：Kはネットワーク入力の機能、NOTネットワークの重みやネットワークの出力である

loss(x, class) = -x[class]*K 

。したがって、Kは定数として扱うことができます。

このカスタム条件を実装する最も簡単な方法は何ですか？ updateOutput()機能は簡単ですが、updateGradInput()機能を変更するにはどうすればよいですか？

Answer 1

基本的にあなたの損失関数Lは、入力とターゲットの関数です。したがって、あなたは新しい損失を正しく理解している場合、

loss(input, target) = ClassNLLCriterion(input, target) * K 

があります。そして、あなたは（

updateGradInput[ClassNLLCriterion](input, target) * K + ClassNLLCriterion(input, target) * dK/dinput 

は、そのためだけKのWRTの微分に損失関数の入力を計算する必要がある入力、に対して、あなたの損失関数の導関数を返すupdateGradInputを実装したいあなたはしませんでしたKを計算する公式を与えて）、それを前の行に差し込みます。新しい損失関数はClassNLLCriterionに依存しているため、この損失関数のupdateGradInputとupdateOutputを使用して計算することができます。