I have implemented the model for CIFAR for dateset having 32x32 images size. While testing this model in batch gives correct prediction.
But while giving single image, it is given the following error:
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])
Screenshot (109)1366×768 133 KB ![Screenshot (110)|690x387]