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
相关推荐
程序喵大人6 小时前
解密 Transformer 的核心——注意力机制
人工智能·深度学习·transformer·注意力机制
heroboyluck7 小时前
AI工程师第四课 - 深度学习入门
人工智能·python·深度学习·llama
一楼的猫7 小时前
网文AI辅助怎么过审?番茄200+维度检测与去AI味实战
人工智能·学习·机器学习·自然语言处理·ai写作
Smoothcloud润云7 小时前
训练一个7B参数的大模型需要多少张GPU?大概要跑多久?
人工智能·深度学习·ai·开源·gpu算力
醒过来摸鱼8 小时前
机器学习知识框架:从数据处理到模型部署
人工智能·机器学习
大鹏的NLP博客9 小时前
深度学习模型PyTorch与ONNX数值一致性校验通用技术报告
人工智能·pytorch·深度学习·onnx
lisw059 小时前
智能CAD:理论、算法、应用与展望!
人工智能·机器学习·软件工程
FriendshipT9 小时前
Ultralytics:解读BNContrastiveHead模块
人工智能·pytorch·python·深度学习·目标检测
大学生技术野农10 小时前
基于Yolo26和Grad-CAM可视化融合的深度学习的猕猴桃分级系统-Python源码
python·深度学习·yolo
ShallWeL10 小时前
AUC、F1、召回率怎么选
人工智能·算法·机器学习