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
相关推荐
Quintus五等升4 小时前
深度学习①|线性回归的实现
人工智能·python·深度学习·学习·机器学习·回归·线性回归
汗流浃背了吧,老弟!6 小时前
什么是ResNet
人工智能·深度学习
哥布林学者6 小时前
吴恩达深度学习课程五:自然语言处理 第一周:循环神经网络 (三)语言模型
深度学习·ai
小途软件6 小时前
高校宿舍访客预约管理平台开发
java·人工智能·pytorch·python·深度学习·语言模型
普通网友7 小时前
Bard 的模型压缩技术:在保证性能的前提下如何实现轻量化部署
人工智能·机器学习·bard
捕风捉你7 小时前
【AI转行04】特征工程:治疗 AI 的“学不会”和“想太多”
人工智能·深度学习·机器学习
lixzest8 小时前
C++上位机软件开发入门深度学习
开发语言·c++·深度学习
AI模块工坊8 小时前
【AAAI 2026】即插即用 Spikingformer 重构残差连接,打造高效脉冲 Transformer
深度学习·重构·transformer
棒棒的皮皮8 小时前
【深度学习】YOLO模型评估之指标、可视化曲线分析
人工智能·深度学习·yolo·计算机视觉
2501_9418787410 小时前
在班加罗尔工程实践中构建可持续演进的机器学习平台体系与技术实现分享
人工智能·机器学习