NN設計４（最適化）

optiomを用いて、ニューラルネットを構成する線形層の重みとバイアスの最適化を行うための準備を行う。まず、先頭に次のコード追加する。

code:(先頭).py

import torch.optim as optim

末尾には次のコードを追加する。

code:(追加)nn.py

alpha = 0.01 # 学習率

optimizer = optim.SGD(model.parameters(), lr=alpha)

print('# optimizer:')

print(optimizer)

ニューラルネットワークを構成するインスタンスのparametersメソッドを呼び出すと、そのネットワークがもつ各層の重みやバイアスなどのパラメータが取り出される。これらをSGDによる最適化の対象としている。

/icons/hr.icon

※ ブラウザのバックボタンで戻る