diff --git a/src/mlpack/methods/ann/layer/add_merge_impl.hpp b/src/mlpack/methods/ann/layer/add_merge_impl.hpp index 71437d6859..6d1db08cd5 100644 --- a/src/mlpack/methods/ann/layer/add_merge_impl.hpp +++ b/src/mlpack/methods/ann/layer/add_merge_impl.hpp @@ -115,10 +115,14 @@ template void AddMerge::serialize( Archive& ar, const unsigned int /* version */) { + // Be sure to clear other layers before loading. if (Archive::is_loading::value) network.clear(); ar & BOOST_SERIALIZATION_NVP(network); + ar & BOOST_SERIALIZATION_NVP(model); + ar & BOOST_SERIALIZATION_NVP(run); + ar & BOOST_SERIALIZATION_NVP(ownsLayer); } } // namespace ann diff --git a/src/mlpack/methods/ann/layer/multiply_merge_impl.hpp b/src/mlpack/methods/ann/layer/multiply_merge_impl.hpp index 19d670d113..12ff210292 100644 --- a/src/mlpack/methods/ann/layer/multiply_merge_impl.hpp +++ b/src/mlpack/methods/ann/layer/multiply_merge_impl.hpp @@ -115,10 +115,14 @@ template void MultiplyMerge::serialize( Archive& ar, const unsigned int /* version */) { + // Be sure to clear other layers before loading. if (Archive::is_loading::value) network.clear(); ar & BOOST_SERIALIZATION_NVP(network); + ar & BOOST_SERIALIZATION_NVP(model); + ar & BOOST_SERIALIZATION_NVP(run); + ar & BOOST_SERIALIZATION_NVP(ownsLayer); } } // namespace ann