diff --git a/nntrainer/layers/common_properties.h b/nntrainer/layers/common_properties.h index ca4bd6140f..ad316e9bdd 100644 --- a/nntrainer/layers/common_properties.h +++ b/nntrainer/layers/common_properties.h @@ -865,15 +865,16 @@ class LabelLayer : public Name { struct ActivationTypeInfo { using Enum = nntrainer::ActivationType; static constexpr std::initializer_list EnumList = { - Enum::ACT_TANH, Enum::ACT_SIGMOID, Enum::ACT_RELU, - Enum::ACT_SOFTMAX, Enum::ACT_LEAKY_RELU, Enum::ACT_SWISH, - Enum::ACT_GELU, Enum::ACT_TANH_GELU, Enum::ACT_SIGMOID_GELU, - Enum::ACT_NONE, Enum::ACT_UNKNOWN}; - - static constexpr const char *EnumStr[] = {"tanh", "sigmoid", "relu", - "softmax", "leaky_relu", "swish", - "gelu", "tanh_gelu", "sigmoid_gelu", - "none", "unknown"}; + Enum::ACT_TANH, Enum::ACT_SIGMOID, Enum::ACT_RELU, + Enum::ACT_SWISH, Enum::ACT_GELU, Enum::ACT_TANH_GELU, + Enum::ACT_SIGMOID_GELU, Enum::ACT_SOFTMAX, Enum::ACT_SOFTPLUS, + Enum::ACT_LEAKY_RELU, Enum::ACT_ELU, Enum::ACT_SELU, + Enum::ACT_MISH, Enum::ACT_NONE, Enum::ACT_UNKNOWN}; + + static constexpr const char *EnumStr[] = { + "tanh", "sigmoid", "relu", "swish", "gelu", + "tanh_gelu", "sigmoid_gelu", "softmax", "softplus", "leaky_relu", + "elu", "selu", "mish", "none", "unknown"}; }; /**