Skip to content

supported_activations

Activations supported by CAREamics.

SupportedActivation #

Bases: str, BaseEnum

Supported activation functions.

  • None, no activation will be used.
  • Sigmoid
  • Softmax
  • Tanh
  • ReLU
  • LeakyReLU

All activations are defined in PyTorch.

See: https://pytorch.org/docs/stable/nn.html#loss-functions

Source code in src/careamics/config/support/supported_activations.py
class SupportedActivation(str, BaseEnum):
    """Supported activation functions.

    - None, no activation will be used.
    - Sigmoid
    - Softmax
    - Tanh
    - ReLU
    - LeakyReLU

    All activations are defined in PyTorch.

    See: https://pytorch.org/docs/stable/nn.html#loss-functions
    """

    NONE = "None"
    SIGMOID = "Sigmoid"
    SOFTMAX = "Softmax"
    TANH = "Tanh"
    RELU = "ReLU"
    LEAKYRELU = "LeakyReLU"
    ELU = "ELU"