/pytorch_toolbox

pytorchに便利なツールのリポジトリです

Primary LanguagePython

pytorch_toolbox

pytorchに便利なツールのリポジトリです

focal_loss_multiclass.py

マルチクラス用のfocal lossクラスです。 引数でγ、weight(元論文のアルファに対応)、size_averageが設定できます。

numpy_loader.py

npyファイルで保存したデータセット用のデータローダーです。