1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 배우는 딥러닝 입문
Connected

연습 문제

파라미터 개수 세기

Deep learning 모델은 매우 많은 파라미터로 유명해요. 파라미터가 많을수록 계산 복잡도와 학습 시간이 늘어나기 때문에, 실무자는 자신의 모델에 파라미터가 몇 개인지 알아야 합니다.

이번 연습에서는 먼저 수작업으로 파라미터 개수를 계산해 보세요. 그런 다음 .numel() 메서드를 사용해 결과를 확인합니다.

지침 1/2

undefined XP
    1
    2

질문

아래 모델의 파라미터 개수를 수작업으로 계산해 보세요. 총 몇 개인가요? 콘솔을 계산기로 사용해도 됩니다.

model = nn.Sequential(nn.Linear(9, 4),
                      nn.Linear(4, 2),
                      nn.Linear(2, 1))

가능한 답변