fastreid.layers.activation.Mish Class Reference

Inheritance diagram for fastreid.layers.activation.Mish:

Public Member Functions
	__init__ (self)
	forward (self, x)

Detailed Description

Definition at line 20 of file activation.py.

Constructor & Destructor Documentation

__init__()

fastreid.layers.activation.Mish.__init__

(

self

)

Definition at line 21 of file activation.py.

    def __init__(self):
        super().__init__()
 

Member Function Documentation

forward()

fastreid.layers.activation.Mish.forward	(		self,
			x )

Definition at line 24 of file activation.py.

    def forward(self, x):
        # inlining this saves 1 second per epoch (V100 GPU) vs having a temp x and then returning x(!)
        return x * (torch.tanh(F.softplus(x)))
 
 

---

The documentation for this class was generated from the following file:

• smreid/fastreid/layers/activation.py