本文主要是介绍AttributeError: Can‘t get attribute ‘SiLU‘ on ＜module ‘torch.nn.modules.activation‘——报错简记，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
- 🥇 版权: 本文由【墨理学AI】原创、在CSDN首发、各位大佬、敬请查阅
- 🎉 声明: 作为全网 AI 领域 干货最多的博主之一，❤️ 不负光阴不负卿 ❤️
- ❤️ 如果文章对你有帮助、欢迎一键三连
文章目录
- ❤️ 运行报错如下
- ❤️ 解决方法
- 📘 更多有效教程
- ❤️ 人生苦短， 欢迎和墨理一起学AI
❤️ 运行报错如下
AttributeError: Can't get attribute 'SiLU' on <module 'torch.nn.modules.activation'
运行如下代码时遇到一个报错：
python gen_wts.py yolov5s.pt
报错信息如下：
Traceback (most recent call last):File "gen_wts.py", line 10, in <module>model = torch.load(pt_file, map_location=device)['model'].float() # load to FP32File "/home/墨理/anaconda3/envs/torch15/lib/python3.7/site-packages/torch/serialization.py", line 592, in loadreturn _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)File "/home/墨理/anaconda3/envs/torch15/lib/python3.7/site-packages/torch/serialization.py", line 852, in _loadresult = unpickler.load()
AttributeError: Can't get attribute 'SiLU' on <module 'torch.nn.modules.activation' from '/home/墨理/anaconda3/envs/torch15/lib/python3.7/site-packages/torch/nn/modules/activation.py'>❤️ 解决方法
解决该问题的实际参考链接
https://github.com/ultralytics/yolov5/issues/1849
- 原因是，使用的 conda 环境 pytorch 版本太低了
- 我切换为 安装了 torch 1.8.1 版本的 conda 环境 运行该代码，该错误就消失了
📘 更多有效教程
计算机视觉、超分重建、图像修复、目标检测、模型部署等方向小伙伴可简单交流
此次博文内容难以 以偏概全，如有不恰当的地方，欢迎评论区批评指正
对于即将入行计算机视觉的小伙伴，墨理这里推荐收藏的干货博文目前如下
-
❤️ 深度学习各领域数据集有效整理——持续更新
-
🎉 深度学习模型训练推理——基础环境搭建推荐博文查阅顺序【基础安装—认真帮大家整理了】——【1024专刊】
- 👋 博主的其他几个专栏也有不少干货，不然也不会有数百位本、硕、博研究圈子小伙伴的关注和订阅 欢迎查阅更多专栏博文 ~
- ❤️ 图像风格转换 —— 代码环境搭建 实战教程【关注即可阅】！
- 💜 图像修复-代码环境搭建-知识总结 实战教程 【据说还行】
- 💙 超分重建-代码环境搭建-知识总结 解秘如何让白月光更清晰【脱单神器】
- 💛 YOLO专栏，只有实战，不讲道理 目标检测【建议收藏】！
- ❤️ 实战有效教程之 Get 深度学习入门项目 实战趣学【查阅即懂】！
- 💜 每篇博文都经过磨练捶打，为各位呈现最简洁的技术；各种基础库、高效安装、环境搭建 【一文读懂】
专栏博文保质保量，感谢各位小伙伴点赞、评论鼓励博主的每一分认真创作
❤️ 人生苦短， 欢迎和墨理一起学AI
- 🎉 作为全网 AI 领域 干货最多的博主之一，❤️ 不负光阴不负卿 ❤️
- ❤️ 如果文章对你有帮助、不要忘了、评论鼓励三连
这篇关于AttributeError: Can‘t get attribute ‘SiLU‘ on ＜module ‘torch.nn.modules.activation‘——报错简记的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
