Hello!
I am having this issue in the training when extracting the prediction:
output = model.forward(image)
The shape of the image tensor is: torch.Size([1, 1, 299, 299])
Batch size: 1
Image size: 299, 299
Grayscale
The model is the inception v3
Anyone has any idea why this is happening?
Thank you in advance