问题描述

使用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

缺点：可能会影响其它模型。