diff --git a/src/mlpack/methods/ann/layer/serialization.hpp b/src/mlpack/methods/ann/layer/serialization.hpp index d179ea9167..db3e252af4 100644 --- a/src/mlpack/methods/ann/layer/serialization.hpp +++ b/src/mlpack/methods/ann/layer/serialization.hpp @@ -70,6 +70,7 @@ CEREAL_REGISTER_TYPE(mlpack::SoftmaxType<__VA_ARGS__>); \ CEREAL_REGISTER_TYPE(mlpack::SoftminType<__VA_ARGS__>); \ CEREAL_REGISTER_TYPE(mlpack::HardTanHType<__VA_ARGS__>); \ + CEREAL_REGISTER_MLPACK_LAYERS(arma::mat); #endif