Почему я получаю сообщение об ошибке при разделении данных для Pytorch?

Question

Я использую этот код для разделения данных и проверки:

def load_split_train_test(datadir, valid_size=0.2):
    train_transforms = transforms.Compose([transforms.Resize(224),
                                           transforms.ToTensor()])
    test_transforms = transforms.Compose([transforms.Resize(224),
                                          transforms.ToTensor()])

train_data = datasets.DatasetFolder(dataset, transform=train_transforms)
test_data = datasets.DatasetFolder(dataset, transform=test_transforms)

Я получил эту ошибку, сообщающую об отсутствии одного аргумента:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-253-ee2f89989ed9> in <module>
      4 #                transform=test_transforms)
      5 train_data = datasets.DatasetFolder(dataset,       
----> 6                 transform=train_transforms)
      7 test_data = datasets.DatasetFolder(dataset,
      8                 transform=test_transforms)

TypeError: __init__() missing 1 required positional argument: 'loader'

Что я делаю не так?