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
相关推荐
海边夕阳200622 分钟前
【每天一个AI小知识】:什么是卷积神经网络?
人工智能·经验分享·深度学习·神经网络·机器学习·cnn
松岛雾奈.2302 小时前
机器学习--PCA降维算法
人工智能·算法·机器学习
5***79002 小时前
机器学习社区机器学习社区:推动技术进步与创新的引擎
人工智能·机器学习
robator3 小时前
label-studio 使用机器学习后端进行预标注
人工智能·机器学习
limenga1024 小时前
支持向量机(SVM)深度解析:理解最大间隔原理
算法·机器学习·支持向量机
爱跑步的george4 小时前
时间序列预测的“诸神之战”:树模型 (XGB/LGBM) vs 深度学习(Transformer/RNN) 的工业界大对决
深度学习
PyAIGCMaster4 小时前
如何编译一个apk,我是新手
深度学习·学习
清云逸仙5 小时前
什么是AI领域的Prompt
人工智能·深度学习·机器学习·prompt
木头左5 小时前
迁移学习的基于股票数据预训练加速
人工智能·机器学习·迁移学习
背影疾风5 小时前
基于深度学习的3D点云特征提取方法梳理
人工智能·深度学习·3d·自动驾驶