p.197 input_volume_size 문의

[teddylee777]

저자님 안녕하세요?

책 정말 잘 보고 있습니다.

p.197의 상단 2번째 줄에서

(784 - 3 + (2*1)/1 + 1 = 784로 나와있는데요.

Fashion MNIST의 입력은 (1, 28, 28) 즉 가로세로 (28, 28) 1 채널 형태로 입력되니,

W=28 이되고 위의 수식은

(28 - 3 + (2*1)/1 + 1 = 28 로 수정되어야 하지 않나 싶습니다.

계산 결과를 적용하면 [32, 784, 784]가 된다고 나와 있는데요.

이 부분은 [32, 28, 28] 로 정정되어야 하지 않을까 싶습니다.

사실, Conv2d의 출력 형태는 맨 앞에 batch_size가 붙기 때문에

[1, 32, 28, 28]이 맞는 표현일 수도 있을 것 같습니다.

혹시 제가 잘못 알고 있는 부분일 수도 있는데요. 한 번 확인 부탁 드립니다 :)

좋은 책 집필해 주셔서 다시 한 번 감사드립니다.

[sonstory]

안녕하세요! 같은 궁금증이 생겨서 확인해 봤는데 질문자 분의 말씀이 맞은신거 같다는 생각이 듭니다. 여기에 추가적으로 궁금한 부분이 있어서 글 남깁니다.

1. batch_size는 앞서 100으로 설정했으니 크기는 [100, 32, 28, 28]로 나오는 것이 맞지 않나요??
2. 또한 CNN 모델에서 fc1 layer의 in_features를 64x6x6으로 설정했는데, 계산상 64x7x7로 설정하는 것이 맞지 않나요??
   28에서 layer1에서 conv2d와 maxpool2d를 거쳐 14가 되고, layer2에서 conv2d와 maxpool2d를 거쳐 7이 되는거 같은데, 6이 맞는지 궁금합니다.

감사합니다.

[sonstory]

padding=0인 것을 반영을 안했습니다. 결론적으로 64x6x6이 맞습니다!