Update adaptive_mean_pooling_impl.hpp

Fixed Serialization issue.
This commit is contained in:
Shubham Agrawal
2022-06-26 17:28:27 +05:30
committed by GitHub
parent 5fd7fc3595
commit a09b3612fa
@@ -146,6 +146,8 @@ void AdaptiveMeanPoolingType<MatType>::serialize(
ar(cereal::base_class<Layer<MatType>>(this));
ar(CEREAL_NVP(poolingLayer));
ar(CEREAL_NVP(outputWidth));
ar(CEREAL_NVP(outputHeight));
}
} // namespace ann