AlexNet 공부 일기

 Image classification 모델을 만들고 싶었다. 그런데 내가 직접 layer를 짜는 것은 정확도가 낮았다. 

그러면 다른 사람이 만든 것을 써야지 생각했다. 찾아보던 중 AlexNet이라는 모델이 눈에 들어왔다. 그래도 그냥 가져다 쓰는 것보다는 어떻게 구성되어 있는지 보고 쓰는 게 좋을 것 같다 생각하여 구조를 살펴봤다. 

AlexNet의 구조는 여기에 올라와있다. 참고로 나는 Convolutional layer, Linear layer, MaxPooling, Avgpooling, ReLU, Flatten등등 막연하게만 알고 있었다. 구조를 보다 보니 이러한 개념들을 정확하게 알아야 따라갈 수 있었다. 

메모장에 볼펜으로 적어나갔다. 처음에는 시간이 많이 걸렸다. 하지만 요령이 조금 생기기 시작했다. 마침내 어느정도 이해가 가능했다. AlexNet의 모델이 어떻게 구성되었는지에 대하여 약간 알고 나니, 마음이 편해졌다. 

 

1