bb936f758a92f6b905aa3c3566e64648cc59cd8d,models.py,Mish,forward,#Mish#Any#,129

Before Change

        super().__init__()

    def forward(self, x):
        return x * torch.tanh(F.softplus(x))


class YOLOLayer(nn.Module):
    def __init__(self, anchors, nc, img_size, yolo_index, arc):

After Change

        super().__init__()

    def forward(self, x):
        return x.mul_(F.softplus(x).tanh())


class YOLOLayer(nn.Module):

Instances