IT博客汇 | 执行torch.load(模型名称, map_location=\'cpu\')报错：from torchvision.transforms.functional_tensor import rgb_to_grayscale

执行torch.load(模型名称, map_location=\'cpu\')报错：from torchvision.transforms.functional_tensor import rgb_to_grayscale

小王发表于 2024-06-11 17:06:00

解决办法：

打开这个文件：

/usr/local/lib/python3.9/site-packages/basicsr/data/degradations.py

第8行：

from torchvision.transforms.functional_tensor import rgb_to_grayscale

改为：

from torchvision.transforms.functional import rgb_to_grayscale

小王 2024-06-12 01:06 发表评论