doiken23 / pytorch_toolbox

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch_toolbox

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

focal_loss_multiclass.py

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

numpy_loader.py

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

About

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


Languages

Language:Python 100.0%