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
相关推荐
richard_first1 小时前
Transformer与大语言模型:第11章 Encoder 是如何理解句子的
深度学习·语言模型·transformer
格林威2 小时前
C#图像分块处理:图像按行或按块(Tile)切分,多个 CPU 核心同时处理不同的区域
开发语言·图像处理·人工智能·机器学习·计算机视觉·c#·工业相机
Sinclair2 小时前
从神经网络到文件加密:一次关于 SIREN、ARX 与密码安全性的实验探索
人工智能·机器学习
单友人2 小时前
HMHA-RYST 算法技能包 + 聊天记录导出器
人工智能·神经网络·机器学习·语言模型·自动化
sel_92 小时前
【多轮对话论文导读(七)】多轮对话论文阅读笔记:从数据生成、用户模拟到上下文重构与长期记忆
论文阅读·人工智能·笔记·深度学习·算法·语言模型·自然语言处理
Zach_菠萝侠2 小时前
【deepseek harness研究】进化方向10:对外编程接口面 思考、设计与实现
人工智能·深度学习·deepseek
宣宣猪的小花园.3 小时前
【控制理论】系统为什么会“有记忆”:从输入输出理解动态过程
人工智能·嵌入式硬件·机器学习
ai小陈3 小时前
PyTorch多GPU分布式训练实战:从单卡脚本迁移到DDP
服务器·人工智能·pytorch·分布式·深度学习·ai·gpu算力
小小帅呀3 小时前
学习 VLA 第 4 天:Transformer 原理与复现实战
深度学习·学习·transformer
EW Frontier3 小时前
CMA-Former:基于级联多头注意力 Transformer 的 DOA 波达方向估计(PyTorch 实现)
pytorch·深度学习·transformer·doa估计