PyTorch convert function for op ‘hardswish‘ not implemented.

AI算法网奇

已于 2022-09-04 15:48:03 修改

阅读量2.7k

点赞数

分类专栏： pytorch知识宝典文章标签： pytorch python 深度学习

于 2020-10-26 22:08:17 首次发布

本文链接：https://blog.csdn.net/jacke121/article/details/109300271

版权

pytorch知识宝典专栏收录该内容

499 篇文章 227 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客主要讨论了在PyTorch中遇到'hardswish'操作未实现的问题，该问题源于torch版本不兼容。作者提到了两个解决方法：一是升级PyTorch到1.6.0；二是提供了手动实现hardswish的代码片段，包括HardSigmoid和SiLU等激活函数的实现。

摘要由CSDN通过智能技术生成

PyTorch convert function for op 'hardswish' not implemented.

一般是torch版本不对，应该是torch1.6.0，但是我的是1.6.0 cu101

解决方法1：

升级pytorch

解决方法2：

Hardswish代码：

class Hardswish(nn.Module):  # export-friendly version of nn.Hardswish()
    @staticmethod
    def forward(x):
        # return x * F.hardsigmoid(x)  # for torchscript and CoreML
        return x * F.hardtanh(x + 3, 0., 6.) / 6.  # for torchscript, CoreML and ONNX