but i cannot show the images by :
test_img, test_lb = next(iter(full_dataset))
print(test_img[0].shape)
plt.imshow(test_img[0].permute(2,1,0))

which is my main problem , is it becuase they are nifty fromt