AlexNet의 첫 번째 합성곱층 커널의 크기는 11×11×3이며, 스트라이드를 4로 적용하여 특성 맵을 96개 생성하기 때문에 55×55×96의 출력을 갖습니다. 첫 번째 계층을 거치면서 GPU-1에서는 주로 컬러와 상관없는 정보를 추출하기 위한 커널이 학습되고, GPU-2에서는 주로 컬러와 관련된 정보를 추출하기 위한 커널이 학습됩니다.

각 GPU의 결과가 다음 그림과 같이 나타납니다.

▲ 그림 6-10 AlexNet GPU-1·2 적용 결과

이제 파이토치 코드로 AlexNet에 대해 살펴봅시다. 먼저 필요한 라이브러리를 호출합니다. 전반적인 코드는 LeNet과 크게 다르지 않기 때문에 주로 모델을 구성하는 네트워크가 어떻게 차이가 나는지 위주로 살펴보면 좋습니다.