7. 필수과제 : Optimization Assignment

7. 필수과제 : Optimization Assignment

nn.Linear( FILL IN HERE ) : 선형레이어 선언 들어가야할 파라미터로는 Input dimension 크기와 Output dimension 크기

여기서는 List에 nn.Linear를 모두 담아두고, 후에 Sequential로 레이어를 쌓아 모델을 완성함

이전 히든 노드 크기, 다음 히든 노드 크기를 들어가야할 파라미터로 넘겨줌

여기서 첫번째 선형 레이어는 이전 히든 노드의 크기가 입력데이터가 됨

optm_sgd = optim.SGD( FILL IN HERE ) : SGD 옵티마이저 선언

optm_momentum = optim.SGD( FILL IN HERE ) : Momentum 옵티마이저 선언

from http://14blacktea.tistory.com/98 by ccl(A) rewrite - 2021-08-11 00:25:51