TypeError: Invalid shape (1, 224, 224, 3) for image data code example
Example: TypeError: Invalid shape (3, 32, 32) for image data
plt.imshow(data[0][0].view(32,32,3))
plt.show()
plt.imshow(data[0][0].view(32,32,3))
plt.show()