Fix the issue of the missing val.txt (#163)

classification/dataset/build.py

@@ -12,7 +12,7 @@ import torch.distributed as dist
 from timm.data import Mixup, create_transform
 from torchvision import transforms
 
-from .cached_image_folder import ImageCephDataset
+from .cached_image_folder import CachedImageFolder, ImageCephDataset
 from .samplers import NodeDistributedSampler, SubsetRandomSampler
 
 try:
@@ -214,7 +214,7 @@ def build_dataset(split, config):
                                        on_memory=config.DATA.IMG_ON_MEMORY)
         elif prefix == 'val':
             root = os.path.join(config.DATA.DATA_PATH, 'val')
-            dataset = ImageCephDataset(root, 'val', transform=transform)
+            dataset = CachedImageFolder(root, transform=transform)
         nb_classes = 1000
     elif config.DATA.DATASET == 'imagenet22K':
         if prefix == 'train':

classification/train_in1k.sh

@@ -24,5 +24,5 @@ PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
     --cfg ${CONFIG} \
     --accumulation-steps 1 \
     --local-rank 0 \
-    --data-path /mnt/lustre/share/images \
+    --data-path data/imagenet \
     --output work_dirs ${@:4}

classification/train_in1k_deepspeed.sh

@@ -23,5 +23,5 @@ PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
     python -u main_deepspeed.py \
     --cfg ${CONFIG} \
     --local-rank 0 \
-    --data-path /mnt/lustre/share/images \
+    --data-path data/imagenet \
     --output work_dirs_deepspeed ${@:4}