[解决方法汇总] pytorch报错：‘GELU‘ object has no attribute ‘approximate‘
使用yolo v5测试模型时，报错。
·
问题描述
使用yolo v5测试模型时，报错
AttributeError: ‘GELU’ object has no attribute ‘approximate’
问题原因
旧版的pytorch中GELU函数没有参数 ‘approximate’。
解决方法
1.为模块添加属性
yolo v5源码（models/experimental.py）中有相似的处理，为GELU模块添加approximate属性，并赋给默认值。
注意：由GELU源码可知，approximate的默认参数是字符串‘none’，而不是None。
2.降低pytorch版本
参考链接：AttributeError: ‘GELU’ object has no attribute ‘approximate’
或者
缺点：操作麻烦。
3.修改pytorch源码
找到GELU的源码，将下图中红框的部分删掉。
参考链接：解决:GELU object has no attribute approximately
缺点：可能会影响其它模型。
华为开发者空间，是为全球开发者打造的专属开发空间，汇聚了华为优质开发资源及工具，致力于让每一位开发者拥有一台云主机，基于华为根生态开发、创新。
更多推荐
7
0- 0
已为社区贡献1条内容
所有评论(0)