𝄢 𝄢

﹡ 이미지 다중분류 문제에서 CNN을 사용하면 convolution layer 와 ReLU, Batchnorm2d, MaxPool2d를 반복하다가 마지막에는 fully connected layer를 통과해서 softmax를 얻어 분류를 해주어야 한다. 예를 들어 CIFAR10 데이터셋으로 10가지를 분류하는 네트워크를 만든다고 하자. 통상적인 네트웍에서는 CNN 레이어를 통과해서 다양한 특징맵을 뽑아내고 풀링을 해서 사이즈를 반으로 줄인다. (Maxpooling을 한다고 해서 채널수가 줄지는 않는다. 채널별로 각각 풀링해주는 거니까) 충분히 특징맵을 얻었다면 nn.Linear( )를 활용해서 최종적인 출력 노드를 10개로 줄여줘야 할 것이다. 이때 nn.Linear(?, 10)의 ? 부분에 어떤것을 써야..