인프런 커뮤니티 질문&답변
작성한 질문수
nn.Parameter
작성
·
823
0
class로 모델 만드는 설명 해주실 때, linear layer 예제 알려주실 때, 파라미터로 등록하려면 nn.Parameter()로 감싸주라고 하셨는데요,
이후의 코드들을 보면 따로 감싸주는 모습이 안보여서요.
파라미터로 등록할 때 명시적으로 필요 없는 경우가 있는 건지요?
감사합니다.
답변 1
0
잔재미코딩 DaveLee
지식공유자
안녕하세요. 답변 도우미입니다.
nn.Parameter()를 사용하여 tensor를 감싸주는 이유는 해당 tensor를 모델의 파라미터로 등록하기 위해서입니다. 이는 저희가 어떤 식으로 딥러닝이 작동하는지를 raw level 코드로 작성해보며, 이해하기 위함이었고요.
이후에는, PyTorch에서 제공하는 nn.Module를 상속받은 클래스 내에서, nn.Linear, nn.Conv2d 등과 같은 레이어를 사용하면, 이 레이어들이 이미 각자의 파라미터를 nn.Parameter()로 감싸서 가지고 있기 때문에, 별도로 nn.Parameter()를 사용하여 감싸주지 않아도 됩니다.
감사합니다.