cannot import name ‘container_abcs‘ from ‘torch._six‘

进行模型训练的时候,报错:

原因是:torch1.8版本之后container_abcs被移除了,所以我们需要修改timm库中调用container_abcs的调用方式:

timm库中调用container_abcs的文件是"~\Conda\envs\pytorch\Lib\site-packages\timm\models\layers\helpers.py"

文件的调用程序如图:

我们对其进行修改,根据Pytorch的版本来选择不同的调用方式(把原来那句话替换掉或者注释掉)

python 复制代码
# 根据torch版本自动选择不同的导入方式
import torch
TORCH_MAJOR = int(torch.__version__.split('.')[0])
TORCH_MINOR = int(torch.__version__.split('.')[1])
if TORCH_MAJOR == 1 and TORCH_MINOR < 8:
    from torch._six import container_abcs
else:
    import collections.abc as container_abcs
相关推荐
极客学术工坊5 小时前
2023年辽宁省数学建模竞赛-B题 数据驱动的水下导航适配区分类预测-基于支持向量机对水下导航适配区分类的研究
机器学习·支持向量机·数学建模·分类
庄周迷蝴蝶5 小时前
旋转位置编码(Rotary Position Embedding,RoPE)
人工智能·机器学习
xier_ran5 小时前
深度学习:RMSprop 优化算法详解
人工智能·深度学习·算法
哥布林学者7 小时前
吴恩达深度学习课程二: 改善深层神经网络 第三周:超参数调整,批量标准化和编程框架(一)超参数调整
深度学习·ai
qy-ll7 小时前
遥感论文学习
人工智能·深度学习·计算机视觉·gan·遥感·栅格化
G31135422737 小时前
深度学习中适合长期租用的高性价比便宜的GPU云服务器有哪些?
服务器·人工智能·深度学习
徐行tag8 小时前
RLS(递归最小二乘)算法详解
人工智能·算法·机器学习
8Qi89 小时前
Stable Diffusion详解
人工智能·深度学习·stable diffusion·图像生成
ChoSeitaku9 小时前
线代强化NO6|矩阵|例题|小结
算法·机器学习·矩阵
carver w9 小时前
transformer 手写数字识别
人工智能·深度学习·transformer