Где находится _DataLoaderIter в pytorch 1.3.1? - PullRequest
2 голосов
/ 20 февраля 2020

Когда я использую pytorch 1.3.1 с python3.7.4, вот так

import torch

from torch.utils.data.dataloader import _DataLoaderIter

Вот ошибка: cannot import name '_DataLoaderIter' from 'torch.utils.data.dataloader'

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

Должен ли я удалить 1.3.1?

Я обнаружил, что _DataLoaderIter находится в dataloader.pyi:

class _DataLoaderIter:
    def __init__(self, loader: DataLoader) -> None:...
    def __len__(self) -> int: ...
    def __iter__(self) -> _DataLoaderIter: ...
    def __next__(self) -> Any: ...

Но я не могу найти его в dataloader.py.

1 Ответ

4 голосов
/ 20 февраля 2020

_DataLoaderIter больше не существует. Этот код является последним, который содержит _DataLoaderIter. Вы можете использовать _SingleProcessDataLoaderIter или _MultiProcessingDataLoaderIter. Я не думаю, что упомянутый вами файл .pyi находится в версии 1.3.1.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...