pytorch_toolbox pytorchに便利なツールのリポジトリです focal_loss_multiclass.py マルチクラス用のfocal lossクラスです。 引数でγ、weight(元論文のアルファに対応)、size_averageが設定できます。 numpy_loader.py npyファイルで保存したデータセット用のデータローダーです。