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
相关推荐
Mr.Lee jack17 小时前
【torch.compile】TorchFX图捕获技术
pytorch
哥布林学者17 小时前
吴恩达深度学习课程四:计算机视觉 第一周:卷积基础知识 课后习题和代码代码实践
深度学习·ai
沃斯堡&蓝鸟17 小时前
DAY22 推断聚类后簇的类型
人工智能·机器学习·聚类
handuoduo123417 小时前
SITAN中avp必要性分析
人工智能·算法·机器学习
【建模先锋】17 小时前
精品数据分享 | 锂电池数据集(七)同济大学电池数据集
深度学习·锂电池剩余寿命预测·锂电池数据集·寿命预测·数据集分享
C嘎嘎嵌入式开发17 小时前
deepseek-r1大模型的本地部署
人工智能·python·神经网络·机器学习
_codemonster17 小时前
AI大模型入门到实战系列(五)上下文嵌入向量(contextualized embedding)
人工智能·深度学习·embedding
Math_teacher_fan18 小时前
第三篇:三角形问题详解
人工智能·学习·机器学习·几何学
一碗白开水一18 小时前
【论文阅读】Denoising Diffusion Probabilistic Models (DDPM)详细解析及公式推导
论文阅读·人工智能·深度学习·算法·机器学习
天上的光18 小时前
机器学习——交叉熵损失函数
人工智能·机器学习