vaeac_check_activation_func: Function that checks the provided activation function
Description
Function that checks the provided activation function
Usage
vaeac_check_activation_func(activation_function)
Value
The function does not return anything.
Arguments
- activation_function
An torch::nn_module()
representing an activation function such as, e.g.,
torch::nn_relu()
(default), torch::nn_leaky_relu()
, torch::nn_selu()
, or torch::nn_sigmoid()
.
Author
Lars Henry Berge Olsen