1. เรียนรู้
    2. /
  2. Courses
    3. /
  3. PyTorchで学ぶIntroduction to Deep Learning
Connected

Exercises

モデルの層を凍結する

事前学習済みの重みを読み込んだあと、新しいタスクに対してモデルをファインチューニングしようとしています。モデルには3つの線形層があります。ただし、データセットが小さいため、このモデルでは最後の線形層だけを学習し、最初の2つの線形層は凍結したいとします。

モデルはすでに作成され、変数modelに入っています。モデルのnamed_parametersメソッドを使って、モデルのパラメータ一覧を取得します。各パラメータには名前が付いており、その名前はx.nameという命名規則の文字列になっています。ここで、xは層のインデックスです。

線形層にはweightとbiasの2つのパラメータがあることを思い出してください。

คำแนะนำ

100 XP
  • if文を使って、パラメータが第1層または第2層のweightかどうかを判定します。
  • このモデルの最初の2層のweightを凍結してください。