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
相关推荐
Sherry Wangs1 小时前
【2025 NeurIPS】Show-o2: Improved Native Unified Multimodal Models
深度学习·多模态·umm
Jerryhut1 小时前
sklearn函数总结四——归一化和标准化
人工智能·python·机器学习·jupyter·sklearn
koo3641 小时前
pytorch深度学习笔记5
pytorch·笔记·深度学习
LUU_793 小时前
Day27 机器学习管道pipeline
人工智能·机器学习
严文文-Chris4 小时前
【半监督学习常见算法】
学习·算法·机器学习
Coding茶水间4 小时前
基于深度学习的35种鸟类检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
祝余Eleanor5 小时前
Day 29 类的定义及参数
人工智能·python·机器学习
ReinaXue5 小时前
跨模态预训练大模型【CLIP】:Contrastive Language–Image Pre-training
图像处理·人工智能·深度学习·计算机视觉·语言模型
福大大架构师每日一题5 小时前
PyTorch v2.9.1 发布:重要 Bug 修复与性能优化详解
人工智能·pytorch·bug
【建模先锋】5 小时前
高效对抗噪声!基于深度残差收缩网络(DRSN)的轴承故障诊断模型
网络·深度学习·信号处理·轴承故障诊断·降噪模型