没有名为“torch._six”的模块可导入队列

Question

我正在尝试应用来自 github 的 RCAN 模型https://github.com/sanghyun-son/EDSR-PyTorch。我正在使用 pytorch 2.0.1。我收到错误：

ModuleNotFoundError: No module named 'torch._six'

尝试导入时：

from torch._six import queue

关于如何在不升级到 torch 1.0 的情况下解决此问题有什么建议吗？

我尝试过以下替代方案：

from six import queue
from multiprocessing import Queue as queue
from multiprocessing.queues import Queue as queue

所有这些都会引发新的错误。我不确定如果队列已被弃用，我什至可以使用什么函数来替换队列，因为我对 pytorch 或模型代码不太熟悉。

Answer 1

快速解决方案：

转到相应的 utils.py 文件并删除以下行：

from taming.data.helper_types import Annotation
from torch._six import string_classes
from torch.utils.data._utils.collate import np_str_obj_array_pattern, default_collate_err_msg_format
from tqdm import tqdm

我希望这能解决您的问题。