2022-01-01から1年間の記事一覧
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…
(チラ裏の日記、こんなことしてたなあとなります) クラスごとのピクセル数に偏りがあるため,ピクセル数が多いクラスほど軽視するよう (多いクラスを正解してもlossが余り減らない)に設定していた.(と思っていた) 損失関数 torch.nn.functional.cross_e…