After creating my trainset as below
trainset = torchvision.datasets.ImageFolder('data/train/', transform=transform)
trainset.classes returns this list
['.ipynb_checkpoints', '0', '1'].
Later when I try to train my model I get the error
IndexError: Target 2 is out of bounds
which I believe is due to my classes being length 3 instead of 2.
How do I fix this safely?