はてなブックマーク

最近訳あってpytorchを入門中です． 自動微分最強と思いつつも，モデルのパラメーターに関する取り扱いについて不明な部分があったので詳しく調べてみました． model.parameters() SGDなどを用いて最適化をする際に，勾配の計算対象とするパラメーターを model.parameters() で渡してあげます． model = MyLinear() optimizer = torch.optim.SGD(model.parameters(), lr=0.01) これだけでパラメーターの最適化をすることができるわけですが，model.parameters()はどうやってモデル内のパラメーターを取得しているのでしょうか？ これを追ってみます． pytorchのバージョンはv0.4.1でお送りします． 大枠 初期化時に，_parameters, _modules という辞書を用意する