search

grooviiee / python_uav

Challenge to Reinforcement learning.
0 stars 0 forks source link

Neural Network를 탈 때, parameter 관리는 어디서 해주는거지? #13

Open grooviiee opened 1 year ago

grooviiee commented 1 year ago

아래와 같은 방식으로 neural network를 실행시키는데, 이 때 hidden layer의 파라미터는 어디서 관리될까? (자동으로 되는지? 다른 관리 변수가 따로 있는지?)

    self.linear_relu_stack = nn.Sequential(
        nn.Linear(28*28, 512),
        nn.ReLU(),
        nn.Linear(512, 512),
        nn.ReLU(),
        nn.Linear(512, 10)
    )
grooviiee commented 1 year ago

Hidden layer의 각 Parameter 들은 nn 구조체(from torch import nn)에서 알아서 관리해주는 듯 하다. Instance가 한 번 생성되었다면, nn API에서 그 Instance의 변수를 알아서 Feed forward, Back propagation 때마다 수정한다.