Pytorch/Practice the code1 CNN 예제 코드 구현하기 PyTorch는 torch.nn 을 포함하여 신경망을 만들고 훈련시키는 것을 도울 수 있도록 모듈과 클래스들을 제공한다. PyTorch에서 신경망 모델은 다음과 같은 순서로 정의된다. 1. nn.Module 을 상속받는 클래스(class)를 생성하여 정의하고 2. __init__ 함수에서 신경망의 계층(layer)들을 정의 3. forward 함수에서 신경망에 데이터를 어떻게 전달할지 지정 nn.Moduel 은 계층, 그리고 output 을 반환하는 forward(input) 메소드를 포함한다. 오늘 예제코드에서는 합성곱을 사용해서 다음과 같은 모델을 정의하고자 한다. 합성곱(convolution)은 커널이나 작은 행렬(matrix)를 통해 가중치를 부여한 이미지의 각 요소를 주변 값과 더하고 입력된 이.. 2022. 8. 23. 이전 1 다음