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 小时前
人工智能治安管控系统
图像处理·人工智能·深度学习·目标检测·计算机视觉·目标跟踪·视觉检测
逄逄不是胖胖7 小时前
《动手学深度学习》-54循环神经网络RNN
人工智能·深度学习
beginner.zs7 小时前
注意力革命:Transformer架构深度解析与全景应用
深度学习·架构·transformer
songyuc8 小时前
【Qwen】DataArguments说明
深度学习·算法·机器学习
2401_841495648 小时前
【机器学习】电商销售额预测实战
人工智能·python·机器学习·数据清洗·矩阵分解·特征可视化·模型训练评估
漂洋过海的鱼儿8 小时前
机器学习-K临近算法(1)
机器学习
辰尘_星启9 小时前
[最优控制]MPC模型预测控制
线性代数·机器学习·机器人·概率论·控制·现代控制
亚里随笔9 小时前
相对优势估计存在偏差——揭示群体相对强化学习中的系统性偏差问题
人工智能·深度学习·机器学习·llm·agentic·rlvr
小白勇闯网安圈9 小时前
神经网络的简单实现
人工智能·深度学习·神经网络
爱吃泡芙的小白白10 小时前
深度学习五大核心指标:F1、Loss、Learning Rate 等的最新演进与应用实战
人工智能·深度学习·训练测试评价指标