Yes you are right. That seems to be the problem.
In reply to the commands
print(test_dataset.classes)
print(test_dataset.class_to_idx)

Yields

[‘ILSVRC2012_img_train’, ‘ILSVRC2012_img_val’]


{‘ILSVRC2012_img_train’: 0, ‘ILSVRC2012_img_val’: 1}

Do i need to have some official dataset ? As i cannot seem to download it when i use this one
torchvision.datasets.ImageNet(root, split='train', download=True, **kwargs)