커스텀 레이어의 max pooling 횟수와 E레이어의 max pooling 횟수가 다르기 때문이다. 이를 고치기 위해서는 커스텀 레이어에 max pooling 횟수를 E레이어와 동일하게 고치거나, nn.Linear(512 * 7 * 7, 4096)의 값을 적절히 수정해 준다.