pytorch

pytorch-lightningのtorchmetricsにある Recallの定義

torchmetrics.readthedocs.io torchmetrics.Recallを使いたかったけど、説明に陽性陰性の定義方法が書いてなかった。ソース辿っていくと metrics/torchmetrics/functional/classification/stat_scores.pyにその定義が書いてあった。 具体的には https://gith…

pytorchのdataloaderとtransformを自作する

pytorchのdataloaderとtransformを自作する 概要 最小構成でのサンプルにより、dataloaderとtransformの自作方法を例示する。 ポイントは前処理を行うtorchvision.transformsに関する部分には、自由に関数を入れることができるというところである。ToTensor(…

pytorchで推論を行うときはmodel.eval()を行おう

pytorchで推論を行いたい。 pytorchでDeep Learningのモデルを組み立てるとき、Batch Normalizationを用いるのはもはや常識だ。しかし、Batch Normalizationを含むモデルで推論を行うとエラーが起きる。 以下のコードをみて欲しい。バッチサイズが1のときに…