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
相关推荐
啦啦啦_99997 小时前
案例之 逻辑回归_电信用户流失预测
算法·机器学习·逻辑回归
我是发哥哈8 小时前
跨AI模型生成视频的五大维度对比:选型避坑指南
大数据·人工智能·学习·机器学习·chatgpt·音视频
机器视觉_Explorer8 小时前
【halcon】编程技巧:鼠标擦除
图像处理·人工智能·深度学习·算法·视觉检测
何陋轩9 小时前
Spring AI实战指南:在Java项目中集成大语言模型
人工智能·后端·机器学习
天天进步201510 小时前
Python全栈项目实战:基于深度学习的语音合成(TTS)系统
开发语言·python·深度学习
香蕉鼠片12 小时前
大模型Function Call
人工智能·深度学习·机器学习·ai
好运的阿财12 小时前
7天没有打开OpenClaw了
python·机器学习·ai·ai编程·openclaw
谙弆悕博士12 小时前
【附Python源码】基于决策树的信用卡欺诈检测实战
python·学习·算法·决策树·机器学习·数据分析·scikit-learn
啦啦啦_999912 小时前
2. ID3决策树 & C4.5决策树
算法·决策树·机器学习
Project_Observer12 小时前
使用Zoho Projects记录工时时间后自动更新项目预算。
开发语言·数据库·人工智能·深度学习·机器学习