nn.Module (1) 썸네일형 리스트형 나만의 신경망 모듈 만들어보기! PyTorch에서 나만의 신경망 계층을 모듈형태로 만들어 보고자 한다. 이 때, 중요한 것이 nn.Module을 상속한 클래스의 정의이다. (nn.Module은 nn.Linear을 포함한 모든 계층의 base class 임) 다음과 같이 nn.Module을 상속한 커스텀 Class를 정의한 후 정의된 선형레이어 Class를 호출하여 쌓아올린뒤 다층신경망을 구축할 수 있는 방법이 하나이고, 또다른 방법은 앞서 정의한 Class를 기반으로 다른 Class를 통해 다층신경망을 구축한 후 마지막에 Class하나만을 호출하여 신경망을 구축하는 방법이다. 딥러닝에 관한 이론적 지식과 PyTorch에 대한 기술적 이해도가 어느정도 갖추어진 상태라면, 자신만의 딥러닝 네트워크를 한번 구축해 보는것은 어떨까? (사실 초.. 이전 1 다음
