PyTorchで自作の損失関数を使う - Qiita

はじめに PyTorchで自作の損失関数の書き方、使い方を説明します。私が使っているPython, PyTorchの環境は以下の通りです。 動作環境 Python 3.7.9 torch 1.6.0+cu101 PyTorch標準の損失関数に倣った書き方 PyTorchに元々あるtorch.nn.MSELossやtorch.nn.CrossEntropyLoss等に倣った書き方です。クラスとして損失関数を定義します。 class CustomLoss(nn.Module): def __init__(self): super().__init__() # 初期化処理 # self.param = ... def forward(self, outputs, targets): ''' outputs: 予測結果(ネットワークの出力) targets: 正解 ''' # 損失の計算 # lo