밑바닥부터 시작하는 딥러닝 5

169

6

장

신경망

그림

6-7

학습 후 모델

그림에서 보듯 데이터에 적합한 모델을 얻을 수 있습니다.

6.3

매개변수와 옵티마이저

신경망을 구현하다 보면 굉장히 많은 매개변수를 다루게 됩니다. 이 매개변수들을 일일이 관리

하는 건 낭비죠. 그래서 파이토치는 매개변수들을 자동으로 관리해주는 메커니즘을 제공합니

다. 예를 들어 다음 클래스들을 이용하면 매개변수들을 간편하게 관리할 수 있습니다.

●

torch

.

nn

.

Parameter

●

torch

.

nn

.

Module

●

torch

.

optim

.

SGD

이번에는 앞 절에서 푼 문제를 여기 세 클래스를 사용하여 다시 작성하겠습니다.