Colab help, [Errno 2] No such file or directory

Hello all,
I’m new to roboflow and was trying out ViT project from How to Train the Hugging Face Vision Transformer On a Custom Dataset, but faced some simple errors that I’m not sure how to get rid of. Can anyone help to solve this Errno2?

Also, the next part has a directory issue which also not sure how to edit.

import torchvision
from torchvision.transforms import ToTensor

train_ds = torchvision.datasets.ImageFolder(‘/content/train/’, transform=ToTensor())
valid_ds = torchvision.datasets.ImageFolder(‘/content/valid/’, transform=ToTensor())
test_ds = torchvision.datasets.ImageFolder(‘/content/test/’, transform=ToTensor())

FileNotFoundError Traceback (most recent call last)
2 from torchvision.transforms import ToTensor
----> 4 train_ds = torchvision.datasets.ImageFolder(‘/content/train/’, transform=ToTensor())
5 valid_ds = torchvision.datasets.ImageFolder(‘/content/valid/’, transform=ToTensor())
6 test_ds = torchvision.datasets.ImageFolder(‘/content/test/’, transform=ToTensor())

3 frames
/usr/local/lib/python3.8/dist-packages/torchvision/datasets/ in find_classes(directory)
38 See :class:DatasetFolder for details.
39 “”"
—> 40 classes = sorted( for entry in os.scandir(directory) if entry.is_dir())
41 if not classes:
42 raise FileNotFoundError(f"Couldn’t find any class folder in {directory}.")

FileNotFoundError: [Errno 2] No such file or directory: ‘/content/train/’

Thank you,

Hi @Sun_Ho_Ro :wave: Do you experience that problem in our original notebook?

Yes its the original colab noteobok

We are working on a solution. In the meantime, you can use a workaround and install roboflow==0.2.29