Ошибка с _DataLoaderIter в torch.utils.data.dataloader

Question

Я хочу запустить код, который должен импортировать _DataLoaderIter из torch.utils.data.dataloader.Проверяя исходный код для класса загрузчика данных, этот метод существует.Однако я получаю сообщение об ошибке:

Traceback (последний вызов был последним):

Файл "main.py", строка 4, в

данные импорта

Файл "D: \ Hyperspectral Data \ RCAN \ RCAN_TrainCode \ code \ data \ __ init__.py", строка 3, в модуле

из импорта данных из MSDataLoader

Файл "D: \ Hyperspectral Data \ RCAN \ RCAN_TrainCode \ code \ dataloader.py ", строка 14, в модуле

из torch.utils.data.dataloader import _DataLoaderIter

ImportError: невозможно импортировать имя '_DataLoaderIter'

Почему это происходит?

Answer 1

Создайте новую виртуальную среду и установите Torch0.4.0 для этого конкретного репо.У меня была такая же проблема, и переустановка Torch v0.4.0 сделала свое дело для меня

Answer 2

Ваш комментарий отвечает на вопрос: _DataLoaderIter есть в 1.0.0 (для которого вы связываете документацию), но не в 0.3.1, так как вы можете проверить здесь - его имя не имеетprevious _.

Это пример из учебника, почему плохая идея получить доступ к закрытым классам / функциям других пакетов (обычно с префиксом подчеркивания) - у вас нет нулевых гарантий стабильности их реализации иповедение.Если вам нужен их код, обычно лучше скопировать и вставить код в ваш собственный файл, потому что там, по крайней мере, гарантируется, что он не изменится между обновлениями и исправлениями ошибок до torch.