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
相关推荐
嘿丨嘿21 分钟前
VLA 入门(六):VLA 如何进行强化学习后训练?
人工智能·python·深度学习·机器人
larance5 小时前
机器学习特征预处理之删移除无关特征
人工智能·机器学习·数据挖掘
薛定e的猫咪9 小时前
【模型推理】深度学习模型部署核心:算子Lower拆解、计算图优化与PNNX源码解析
人工智能·深度学习
lisw059 小时前
【计算机科学技术】从晶圆制备到芯片出厂的流程
人工智能·机器学习·制造
lisw059 小时前
【计算机科学技术】晶圆是什么回事?与芯片的区别与联系!
机器学习·制造
Helen_cai11 小时前
HarmonyOS ArkTS 实战:实现一个校园门禁与访客预约应用
深度学习·华为·harmonyos
All The Way North-12 小时前
【TorchMetrics精通系列①】核心设计哲学 + Accuracy 超详解
机器学习·分类·模型评估·accuracy·准确率·评估指标·torchmetrics
2zcode12 小时前
项目文档:基于MATLAB深度卷积特征的多曝光图像自适应融合系统设计与实现
深度学习·计算机视觉·matlab
湘美书院--湘美谈教育13 小时前
湘美谈教育湘美书院成功学系列:标准即是文明,AI时代的走向
大数据·人工智能·深度学习·机器学习·生活