probly.method.prior_network.torch.append_activation_torch

probly.method.prior_network.torch.append_activation_torch(obj: Module) Sequential[source]

Append exp so the model outputs Dirichlet alpha based on [MG18].

Unlike Sensoy’s softplus + 1 parameterization, exp allows alpha values below 1, which the Prior Networks paper uses for very flat Dirichlets on out-of-distribution inputs.