No such file or directory: 'CIFAR-FS/train/'

[ry-jojo]

Hi there,
Thanks for sharing the codebase of such an awesome paper.
I am facing with an dataset-related error as follows when running ./scripts/pretrain_teacher.sh:

File "code/pretrain_teacher.py", line 211, in main
dataloader, num_classes = load_data(root=args.dataset, batch_size=batch_size)
File "code/pretrain_teacher.py", line 38, in load_data
trainset = torchvision.datasets.ImageFolder(f'{root}/train/', transform=transform) ## Training Datasets
"No such file or directory: 'CIFAR-FS/train/' "

My CIFAR-FS folder does not have the subfolder: train after running prepare.py.

Could you please give some suggestions? Thanks in advance.