yiskw note

機械学習やプログラミングについて気まぐれで書きます

2021-03-02から1日間の記事一覧

pytorchで全結合層以外の重みを固定して学習し直す

概要 pytorchで転移学習を行う際に,全結合層以外の重みを固定するということをよくやるのですが,毎回やり方を忘れて調べていたので,備忘録としてこちらに残しておきます.