diff --git a/python_module/megengine/data/dataset/vision/imagenet.py b/python_module/megengine/data/dataset/vision/imagenet.py index 82fcaca10d954e46a4bdec8ef8b9aebe0161261d..3242cecf46b34ce6fc6a29c342654558af41ed5d 100644 --- a/python_module/megengine/data/dataset/vision/imagenet.py +++ b/python_module/megengine/data/dataset/vision/imagenet.py @@ -224,7 +224,7 @@ class ImageNet(ImageFolder): untar(path, os.path.splitext(path)[0], remove=True) def _prepare_devkit(self): - raw_filename, checksum = self.raw_file_meta["val"] + raw_filename, checksum = self.raw_file_meta["devkit"] raw_file = os.path.join(self.root, raw_filename) logger.info("checksum devkit tar file {} ..".format(raw_file)) assert calculate_md5(raw_file) == checksum, \