Solved by kenmikanmi in post #2 
              
              
                Hi, in the forward() function, you can find that the self.pool() is applied twice. 
def forward(self, x):
    x = self.pool(F.relu(self.conv1(x)))
    x = self.pool(F.relu(self.conv2(x)))      # <- Here
    x = x.view(-1, 16 * 5 * 5)
    x = F.relu(self.fc1(x))
    x = F.relu(self.fc2(x))
    x = se…