Torchの損失機能をカスタマイズするにはどうすればよいですか？

Question

私は現在トーチ7を使用していますが、特にクロスエントロピーエラー機能をカスタマイズする必要があります。

クロスエントロピーエラー関数にいくつかのパラメータを追加することを考えていますが、どの部分を修正するべきかを見つけることができませんでした。

私はCrossEntropyCriterion.luaを見ましたが、このファイルにはまだ何も表示されません。

誰でも私の方程式はどこですか？どのファイルを変更する必要がありますか？

Answer 1

損失関数をカスタマイズするには、__init,updateOutputおよびupdateGradInputのメソッドを変更する必要があります。

• __init
  はあなたが:backward()を使用する場合
• updateGradInputが呼び出されます、あなたの基準で:forward()メソッドを使用する場合
• updateOutputが呼ばれるクラスの初期化関数であり、それはあなたの基準の勾配である

カスタマイズされた基準の構造は次のようになります。

local yourCriterion, parent = torch.class('nn.yourCriterion', 'nn.Criterion') 

function yourCriterion:__init(your_parameters): 
    parent.__init(self) 
    ... (you can add as many parameters as you want to your criterion 
     and give them the name your prefer) 
    self.parameters = your_parameters 


function yourCriterion:updateOutput(input) 
    ... (your criterion code here) 
    return value_of_the_criterion 

function yourCriterion:updateGradInput(input): 
    ... (your criterion gradient code here) 
    return gradient 

[EDIT]：あなたがここにクロスエントロピー基準のコードを見つけることができますhttps://github.com/torch/nn/blob/master/CrossEntropyCriterion.lua

[トーチに私のカスタムの損失関数を追加]の