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
相关推荐
海边夕阳20068 小时前
【每天一个AI小知识】:什么是图神经网络?
人工智能·经验分享·深度学习·神经网络·机器学习
Brduino脑机接口技术答疑8 小时前
TDCA 算法在 SSVEP-BCI 中的时间戳技术要求与工程实现
人工智能·深度学习·机器学习·脑机接口·ssvep
musk12128 小时前
english-12-word-25-12-25 , 单词 legacy Linear scenes 深度学习知识卡片
人工智能·深度学习
啊阿狸不会拉杆8 小时前
《数字图像处理》实验2-空间域灰度变换与滤波处理
图像处理·人工智能·机器学习·计算机视觉·数字图像处理
EniacCheng8 小时前
贝叶斯定理
人工智能·机器学习·概率论
Melody20509 小时前
mmcv的依赖安装教程
人工智能·深度学习
过期的秋刀鱼!9 小时前
week3-机器学习-逻辑回归模型介绍和决策边界
人工智能·机器学习·逻辑回归
Godspeed Zhao9 小时前
自动驾驶中的传感器技术84——Sensor Fusion(7)
人工智能·机器学习·自动驾驶
摸鱼仙人~9 小时前
多种类型Agent 工具调用机制讲解
人工智能·pytorch·agent
JoannaJuanCV9 小时前
自动驾驶—CARLA仿真(28)地图与导航(Maps and navigation)
人工智能·机器学习·自动驾驶·carla