Skip to content

Commit f30cf42

Browse files
authored
Add CELU activation function (#851)
1 parent ad863fd commit f30cf42

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

physicsnemo/models/layers/activations.py

+1
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,7 @@ def forward(self, inputs):
176176
"prelu": nn.PReLU,
177177
"relu6": nn.ReLU6,
178178
"elu": nn.ELU,
179+
"celu": (nn.CELU, {"alpha": 1.0}),
179180
"selu": nn.SELU,
180181
"silu": nn.SiLU,
181182
"gelu": nn.GELU,

0 commit comments

Comments
 (0)