Hi all,
I am new to python programming in deep learning. And I was executing some basic code of classifying the MNIST dataset, but I keep encountering this error:
IndexError: too many indices for tensor of dimension 0
Each time i execute this part of my code:
dataiter = iter(training_loader)
images, labels = dataiter.next()
fig = plt.figure(figsize = (25, 4))
for idx in np.arange(20):
ax = fig.add_subplot(2, 10, idx+1)
plt.imshow(im_convert(images[idx]))
ax.set_title([labels[idx].item()])
I need HELP with this!