MOB-LOG

モブおじの記録 (Programming, 統計・機械学習)

2022-12-01から1ヶ月間の記事一覧

torch.nn.functional.cross_entropyで、weight指定してlabel_smoothing>0 のときにOut of bounds エラー(ignore_index が機能していない疑惑)

torch.nn.functional.cross_entropy(Input, Target, ...) において、weight (size=[tex:(C,)])を指定、ignore_index をクラスサイズC以外のインデックスに指定して (e.g. -100, -255)、label_smoothing > 0, reduction='mean'とするときだけ Out of bounds e…

(メモ) torch.nn.functional.cross_entropy, 損失計算のweightの設定

(チラ裏の日記、こんなことしてたなあとなります) クラスごとのピクセル数に偏りがあるため,ピクセル数が多いクラスほど軽視するよう (多いクラスを正解してもlossが余り減らない)に設定していた.(と思っていた) 損失関数 torch.nn.functional.cross_e…