deep-learning-for-image-processing 高版本的torch报错未发现torch.

System information

Have I writtenTraceback (most recent call last): File "C:\Work\Pycharm\faster_rcnn\train_mobilenetv2.py", line 11, in from train_utils import GroupedBatchSampler, create_aspect_ratio_groups File "C:\Work\Pycharm\faster_rcnn\train_utils_init_.py", line 4, in from .coco_eval import CocoEvaluator File "C:\Work\Pycharm\faster_rcnn\train_utils\coco_eval.py", line 7, in import torch._six ModuleNotFoundError: No module named 'torch._six'

我的torch版本是2.0.1

Jul 18 '23 08:07 water107

我也遇到了同样的问题，请问你解决了吗

Aug 09 '23 05:08 mjc-ddj

import torch._six我直接将这一段注释掉了，debug并没有报错

Aug 09 '23 09:08 mjc-ddj

有两种方式

# 把import语句注释掉，并添加string_classes = str
# from torch._six import *
string_classes = str

# 然后再后面找到使用__six函数的地方，把torch.__six.string_classes改为string_classes，如：
if isinstance(resFile, string_classes):

# 也可以直接改成：
if isinstance(resFile, str):

Aug 11 '23 14:08 linwqwhu

高版本的torch报错未发现torch._six