I have resolved. It was regarding the size of the images: The network was trained on torch.Size([1, 3, 32, 32]) and validation imageswere size of torch.Size([1, 3, 64, 64]). This means that neural network treated validation images as batches of 4 images and that’s reason for the output that dimensions.