noether.modeling.modules.activations

Classes

Activation

Supported activation functions.

Module Contents

class noether.modeling.modules.activations.Activation(*args, **kwds)

Bases: enum.Enum

Supported activation functions.

GELU
TANH
SIGMOID
RELU
LEAKY_RELU
SOFTPLUS
ELU
SILU
build()

Create a new instance of the activation module.

Return type:

torch.nn.Module