Data_loader приводит к ошибке вызова «set»

#pytorch

Вопрос:

 train_data_path = '/content/gdrive/MyDrive/dataset/train'

train_data = datasets.ImageFolder(root=train_data_path, transform=data_transforms)
train_data_loader = torch.utils.data.DataLoader(train_data, batch_size=8, shuffle=True)



for xs, ts in train_data_loader:
  print(xs)
  print(ts) 
 

В настоящее время я пытаюсь перетасовать и преобразовать изображения в папке в набор данных тензоров. Одна из проблем, с которой я сталкиваюсь, заключается в том, что это приводит к установленной вызываемой ошибке, и я очень смущен тем, почему это происходит. Указанный путь содержит вложенную папку с изображениями.