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
相关推荐
xwill*18 小时前
分词器(Tokenizer)-sentencepiece(把训练语料中的字符自动组合成一个最优的子词(subword)集合。)
开发语言·pytorch·python
哥布林学者20 小时前
吴恩达深度学习课程四:计算机视觉 第一周:卷积基础知识(一)图像处理基础
深度学习·ai
phoenix@Capricornus21 小时前
视觉Transformer(ViT)
人工智能·深度学习·transformer
马踏岛国赏樱花1 天前
Mamba: Linear-Time Sequence Modeling with Selective State Spaces
深度学习
aaaa_a1331 天前
李宏毅——self-attention Transformer
人工智能·深度学习·transformer
cvyoutian1 天前
解决 PyTorch 大型 wheel 下载慢、超时和反复重下的问题
人工智能·pytorch·python
子非鱼9211 天前
3 传统序列模型——RNN
人工智能·rnn·深度学习
万俟淋曦1 天前
【论文速递】2025年第33周(Aug-10-16)(Robotics/Embodied AI/LLM)
人工智能·深度学习·ai·机器人·论文·robotics·具身智能
玦尘、1 天前
《统计学习方法》第6章——逻辑斯谛回归与最大熵模型(上)【学习笔记】
机器学习·回归·学习方法