Hi All,
I am trying to solve a binary image classification problem. I am trying to implement feed forward neural network use.
I am getting a problem - like this.
File “/opt/conda/lib/python3.7/site-packages/torchvision/transforms/functional.py”, line 104, in to_pil_image
raise ValueError(‘pic should be 2/3 dimensional. Got {} dimensions.’.format(pic.ndimension()))
ValueError: pic should be 2/3 dimensional. Got 1 dimensions.
Here is my code.
Please let me know where is the problem . CODE HERE 73
It would be great if someone points the problem at my code location.
I have just joined this forum. Pardon me if this is not the right place to ask these questions.(I’m Beginner in Pytorch use)
Regards
Shravan