nip.parameters.types.ActivationType

Contents

nip.parameters.types.ActivationType#

nip.parameters.types.ActivationType#

Type for the activation function to use.

To add a new activation function, add it to this type, and add the corresponding function to ACTIVATION_CLASSES in nip.utils.torch_modules.

relu

Rectified Linear Unit.

tanh

Hyperbolic Tangent.

sigmoid

Sigmoid.

alias of Literal[‘relu’, ‘tanh’, ‘sigmoid’]