NotImplementedError: когда я пытаюсь создать объект DataLoader в Pytorch - PullRequest
Новая
       26

NotImplementedError: когда я пытаюсь создать объект DataLoader в Pytorch

0 голосов
/ 10 июля 2019

Я создал собственный класс набора данных, который наследуется от класса набора данных PyTorch, для обработки моего пользовательского набора данных, который я уже предварительно обработал.

Когда я пытаюсь создать объект DataLoader, я получаю эту ошибку: 

/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py in __init__(self, dataset, batch_size, shuffle, sampler, batch_sampler, num_workers, collate_fn, pin_memory, drop_last, timeout, worker_init_fn)
    174             if sampler is None:
    175                 if shuffle:
--> 176                     sampler = RandomSampler(dataset)
    177                 else:
    178                     sampler = SequentialSampler(dataset)

/usr/local/lib/python3.6/dist-packages/torch/utils/data/sampler.py in __init__(self, data_source, replacement, num_samples)
     62                              "since a random permute will be performed.")
     63 
---> 64         if not isinstance(self.num_samples, int) or self.num_samples <= 0:
     65             raise ValueError("num_samples should be a positive integer "
     66                              "value, but got num_samples={}".format(self.num_samples))

/usr/local/lib/python3.6/dist-packages/torch/utils/data/sampler.py in num_samples(self)
     70         # dataset size might change at runtime
     71         if self._num_samples is None:
---> 72             return len(self.data_source)
     73         return self._num_samples
     74 

/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataset.py in __len__(self)
     18 
     19     def __len__(self):
---> 20         raise NotImplementedError
     21 
     22     def __add__(self, other):

NotImplementedError:

Итак, сообщение об ошибке касается не реализации функции len () в dataset.py, верно? Но я это реализовал и getitem (), init ().

Как я могу преодолеть это? Спасибо

1 Ответ

0 голосов
/ 10 июля 2019

Убедитесь, что имя правильно в вашем коде.Это должно быть __len__.

...