欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

模型训练技巧——**函数mish

程序员文章站 2022-06-02 14:30:37
...

代码地址:https://github.com/lessw2020/mish

1. mish的公式表达和曲线

   公式表达:Mish=x * tanh(ln(1+e^x))

   曲线表达,如下图所示:

模型训练技巧——**函数mish
Caption

2. mish的pytorch实现

class Mish(torch.nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, x):
        x = x * (torch.tanh(torch.nn.functional.softplus(x)))
        return x

  效果还不错,你也试试吧!