1. Learn
    2. /
  2. Courses
    3. /
  3. PyTorchで学ぶIntroduction to Deep Learning
Connected

Exercise

パラメータ数を数える

Deep learning のモデルは、非常に多くのパラメータを持つことで有名です。パラメータが増えるほど計算量が増え、学習時間も長くなります。実践者は自分のモデルがいくつのパラメータを持つか把握しておく必要があります。

この演習では、まず手計算でパラメータ数を求め、続いて .numel() メソッドを使って結果を検証します。

Instructions 1/2

undefined XP
    1
    2

Question

以下のモデルのパラメータ数を手計算で求めてください。いくつになりますか？コンソールを電卓として使ってかまいません。

model = nn.Sequential(nn.Linear(9, 4),
                      nn.Linear(4, 2),
                      nn.Linear(2, 1))

Possible answers