2022-01-01から1年間の記事一覧

PyTorchのBatch Normalizationがパラメータを固定しても変動する問題の解決法

起こった問題 大規模データセットで学習済みのResNetを用いて転移学習を行うと、ResNetの全てのパラメータに対してrequires_grad = Falseとしているにも関わらず、転移学習の前後でResNetの出力が変化する問題が生じました。 原因 調べると、PyTorchのBatch …

物体検出のDeticを導入する際にトラブった件

はてなブログ開設の1記事目としては全く唐突ですが、物体検出器 Deticを導入するにあたって、起こったトラブルについて調べたことを備忘録がてら残しておこうと思います。 Deticの元論文は以下になります。 arxiv.org 今回起こった問題 githubからDeticをク…