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
相关推荐
GuWenyue9 分钟前
写Agent还要重复封装工具?一套MCP多服务方案,3个能力让AI自动查地图、读写文件、操控浏览器
人工智能·机器学习·开源
一只小bit5 小时前
LangGraph 记忆、人机交互、时间旅行和核心能力
机器学习·langchain·人机交互·langgraph
C^h8 小时前
python函数学习
人工智能·python·机器学习
培之10 小时前
RTX 5090 安装 pytorch3d
人工智能·pytorch·python
大鱼>10 小时前
深入Real-ESRGAN架构:RRDBNet设计精髓与ONNX/TensorRT部署优化
人工智能·深度学习·架构
延卿10 小时前
nnDetection:基于 PyTorch 的目标检测框架
人工智能·pytorch·目标检测
RD_daoyi11 小时前
外链权重暴跌至13%,品牌提及反超传统链接——2026年不做Digital PR,你的独立站等于隐形
运维·网络·学习·机器学习·搜索引擎
2zcode11 小时前
项目文档:基于MATLAB深度卷积神经网络的肺癌CT影像智能检测系统设计与实现
深度学习·matlab·cnn
m沐沐11 小时前
【深度学习】卷积神经网络 数据增强、保存最优模型实现,详细解读
人工智能·python·深度学习·机器学习·cnn·数据增强
硅谷秋水12 小时前
PhyGround:生成式世界模型中的物理推理基准测试
人工智能·深度学习·机器学习·计算机视觉·语言模型