img, label = dataset print('Label (numeric):', label) print('Label (textual):', classes[label]) image_permuted = img.permute((1,2,0)) image_permuted.shape plt.imshow(img)
Can you check in your terminal if there’s an error?
Error message from cmd prompt :
[I 11:18:37.205 NotebookApp] KernelRestarter: restarting kernel (1/5), keep random ports
kernel b9a16664-f1c0-48f1-bb74-cd71bc29cdac restarted
OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program. That is dangerous, since it can degrade performance or cause incorrect results. The best thing to do is to ensure that only a single OpenMP runtime is linked into the process, e.g. by avoiding static linking of the OpenMP runtime in any library. As an unsafe, unsupported, undocumented workaround you can set the environment variable KMP_DUPLICATE_LIB_OK=TRUE to allow the program to continue to execute, but that may cause crashes or silently produce incorrect results. For more information, please see http://www.intel.com/software/products/support/.
Setting Environment variable ‘KMP_DUPLICATE_LIB_OK’ = ‘True’ solved issue.
import os os.environ['KMP_DUPLICATE_LIB_OK']='True'
Thanks, @sidujjain for the hint.