apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #1456 from zoq/merge_run_serialization

Browse Source 
Serialize merge layer parameter
This commit is contained in:
Ryan Curtin
2018-07-05 09:51:23 -04:00
committed by GitHub
parent 77e386860d 328b803b40
commit 32e4446b61
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/add_merge_impl.hpp
+4
View File
@@ -115,10 +115,14 @@ template<typename Archive>
void AddMerge<InputDataType, OutputDataType, CustomLayers...>::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
src/mlpack/methods/ann/layer/multiply_merge_impl.hpp
+4
View File
@@ -115,10 +115,14 @@ template<typename Archive>
void MultiplyMerge<InputDataType, OutputDataType, CustomLayers...>::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