From 3ee8d8ca2bc984e48140f42d26a9ec9bac7d8457 Mon Sep 17 00:00:00 2001 From: Vaibhav Pathak Date: Mon, 27 Mar 2023 19:02:48 +0530 Subject: [PATCH] Fixed error in serialization.hpp --- src/mlpack/methods/ann/layer/serialization.hpp | 1 + 1 file changed, 1 insertion(+) 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