PistonY / torch-toolbox

🛠 Toolbox to extend PyTorch functionalities
BSD 3-Clause "New" or "Revised" License
417 stars 56 forks source link

Added Mish activation function #13

Closed digantamisra98 closed 3 years ago

digantamisra98 commented 3 years ago

Reference: https://www.bmvc2020-conference.com/assets/papers/0928.pdf I didn't implement the backward pass using an autograd function like in the case of swish because of the internal stable differentiation of thresholded softplus.