Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 13:48:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add Sigmoid activation

Browse source
This commit is contained in:
Pierre Chapuis 2023-08-07 16:25:58 +02:00
parent f49bb4f5fd
commit bd49304fc8
2 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/refiners/fluxion/layers/__init__.py
View file

@ -1,4 +1,4 @@
from refiners.fluxion.layers.activations import GLU, SiLU, ReLU, ApproximateGeLU, GeLU
from refiners.fluxion.layers.activations import GLU, SiLU, ReLU, ApproximateGeLU, GeLU, Sigmoid
from refiners.fluxion.layers.norm import LayerNorm, GroupNorm, LayerNorm2d
from refiners.fluxion.layers.attentions import Attention, SelfAttention, SelfAttention2d
from refiners.fluxion.layers.basics import (
@ -44,6 +44,7 @@ __all__ = [
"SiLU",
"ReLU",
"ApproximateGeLU",
"Sigmoid",
"Attention",
"SelfAttention",
"SelfAttention2d",

8
src/refiners/fluxion/layers/activations.py
View file

@ -46,6 +46,14 @@ class ApproximateGeLU(Activation):
return x * sigmoid(1.702 * x)
class Sigmoid(Activation):
def __init__(self) -> None:
super().__init__()
def forward(self, x: Tensor) -> Tensor:
return x.sigmoid()
class GLU(Activation):
"""
Gated Linear Unit activation layer.