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

Merge pull request #1905 from walragatver/mean_pooling

Browse Source 
Fix serialization of glimpse, maxPooling and meanPooling layers.
This commit is contained in:
Marcus Edel
2019-05-31 13:13:50 +02:00
committed by GitHub
parent f4840a7b45 f3318fd0f2
commit c8f52fbd6b
3 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/glimpse_impl.hpp
+3
View File
@@ -224,6 +224,9 @@ void Glimpse<InputDataType, OutputDataType>::serialize(
ar & BOOST_SERIALIZATION_NVP(depth);
ar & BOOST_SERIALIZATION_NVP(scale);
ar & BOOST_SERIALIZATION_NVP(inputWidth);
ar & BOOST_SERIALIZATION_NVP(inputHeight);
ar & BOOST_SERIALIZATION_NVP(outputWidth);
ar & BOOST_SERIALIZATION_NVP(outputHeight);
ar & BOOST_SERIALIZATION_NVP(location);
}
src/mlpack/methods/ann/layer/max_pooling_impl.hpp
+5
View File
@@ -148,6 +148,11 @@ void MaxPooling<InputDataType, OutputDataType>::serialize(
ar & BOOST_SERIALIZATION_NVP(dW);
ar & BOOST_SERIALIZATION_NVP(dH);
ar & BOOST_SERIALIZATION_NVP(batchSize);
ar & BOOST_SERIALIZATION_NVP(floor);
ar & BOOST_SERIALIZATION_NVP(inputWidth);
ar & BOOST_SERIALIZATION_NVP(inputHeight);
ar & BOOST_SERIALIZATION_NVP(outputWidth);
ar & BOOST_SERIALIZATION_NVP(outputHeight);
}
} // namespace ann
src/mlpack/methods/ann/layer/mean_pooling_impl.hpp
+5
View File
@@ -122,6 +122,11 @@ void MeanPooling<InputDataType, OutputDataType>::serialize(
ar & BOOST_SERIALIZATION_NVP(dW);
ar & BOOST_SERIALIZATION_NVP(dH);
ar & BOOST_SERIALIZATION_NVP(batchSize);
ar & BOOST_SERIALIZATION_NVP(floor);
ar & BOOST_SERIALIZATION_NVP(inputWidth);
ar & BOOST_SERIALIZATION_NVP(inputHeight);
ar & BOOST_SERIALIZATION_NVP(outputWidth);
ar & BOOST_SERIALIZATION_NVP(outputHeight);
}
} // namespace ann