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
相关推荐
机器学习之心1 分钟前
TCN-RNN电力负荷预测模型实战
人工智能·rnn·深度学习
宝贝儿好1 分钟前
【NLP】第七章:项目实操案例:智能输入法项目
人工智能·深度学习·神经网络·自然语言处理
郑洁文23 分钟前
基于机器学习的成都市租房数据的分析与应用
人工智能·机器学习·成都市租房数据·成都市租房数据分析
哥布林学者26 分钟前
深度学习进阶(二十九)现代 LLM 的核心架构设计其四:GQA
机器学习·ai
林间码客34 分钟前
《人工智能概论》实验4 知识点复习提纲
人工智能·深度学习·机器学习
chloe233336 分钟前
【动手学深度学习】笔记1:简单的线性回归
笔记·深度学习·线性回归
钓了猫的鱼儿37 分钟前
基于深度学习+AI的秸秆目标检测与预警系统(Python源码+数据集+UI可视化界面+YOLOv11训练结果)
人工智能·深度学习·目标检测
人工智能培训1 小时前
从GPT到开源大模型
人工智能·gpt·深度学习·机器学习·容器·知识图谱
红宝村村长2 小时前
OPD Reverse KL
机器学习
Kobebryant-Manba2 小时前
学习参数管理
pytorch·python·深度学习