enerzyme.models.activation.get_activation_fn

enerzyme.models.activation.get_activation_fn(key: Literal['shifted_softplus', 'swish'], activation_params: Dict[Literal['dim_feature', 'initial_alpha', 'initial_beta', 'learnable'], int | float | bool] = {}) → BaseScaledTemperedActivation