(チラ裏の日記、こんなことしてたなあとなります) クラスごとのピクセル数に偏りがあるため,ピクセル数が多いクラスほど軽視するよう (多いクラスを正解してもlossが余り減らない)に設定していた.(と思っていた) 損失関数 torch.nn.functional.cross_entropy のクラスごとの重みについて CrossEntropyLoss - PyTorch 1.11.0 documentation weight (Tensor, optional) – a manual rescaling weight given to each class. If given, has to b…