Thank you for your question you can do this as followsFirst import the torchvision packagefrom torchvision import datasetsThen change the way of loading imagenet dataet in L156-L164root = ospathjoinco

感谢您的问题，您可以按照以下方式进行操作。 首先导入torchvision包

from torchvision import datasets 然后更改加载Imagenet数据集的方式在L156-L164中

root = os.path.join(config.DATA.DATA_PATH, prefix) dataset = datasets.ImageFolder(root, transform=transform) 我们稍后会更新代码。