if opname == 'hardswish':
relu=_registry[(domain, version)]["relu"]
_registry[(domain, version)][opname] = relu
pytorch 不能导出yolov5的onnx
最新推荐文章于 2024-08-18 18:40:51 发布
这篇博客介绍了如何在指定的域和版本下,将激活函数relu替换为hardswish。这个过程涉及到对Registry的操作,通过更新Registry来实现函数的替换,这对于深度学习模型的性能优化具有重要意义。
摘要由CSDN通过智能技术生成