функция next (iter (..)) генерирует ошибку - PullRequest
Новая
       106

функция next (iter (..)) генерирует ошибку

0 голосов
/ 20 июня 2020

Я применяю эти строки кода в своей модели 

train_data_loader = create_data_loader(df_train, tokenizer, MAX_LEN, BATCH_SIZE)

data =next(iter(train_data_loader))

, но я получил эту ошибку 

TypeError                                 Traceback (most recent call last)
<ipython-input-39-8edd470666f3> in <module>()
----> 1 data =next(iter(train_data_loader))

3 frames
/usr/local/lib/python3.6/dist-packages/torch/_utils.py in reraise(self)
    393             # (https://bugs.python.org/issue2651), so we work around it.
    394             msg = KeyErrorMessage(msg)
--> 395         raise self.exc_type(msg)

TypeError: Caught TypeError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop
    data = fetcher.fetch(index)
  File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "<ipython-input-21-cb3ac03ca3d1>", line 30, in __getitem__
    'targets': torch.tensor(target, dtype=torch.long)
TypeError: new(): invalid data type 'str'

мой набор данных содержит 3 столбца с типами int64, object и object.

Как решить эту проблему?

...