How to Load ImageNet Dataset in PyTorch

Thank you for your question. You can load the ImageNet dataset in PyTorch by following these steps:

1. Import the torchvision package:

from torchvision import datasets

2. Change the way of loading Imagenet dataset in lines L156-L164:

root = os.path.join(config.DATA.DATA_PATH, prefix)
dataset = datasets.ImageFolder(root, transform=transform)

We'll update the code with further explanations and optimizations later. Stay tuned!