Fixed error in serialization.hpp
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user