From a09b3612fa89dd6cb69e90bbbcf5016a8df991ac Mon Sep 17 00:00:00 2001
From: Shubham Agrawal <58412969+shubham1206agra@users.noreply.github.com>
Date: Sun, 26 Jun 2022 17:28:27 +0530
Subject: [PATCH] Update adaptive_mean_pooling_impl.hpp

Fixed Serialization issue.
---
 src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp b/src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp
index f397826bb4..8ad2aad0b8 100644
--- a/src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp
+++ b/src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp
@@ -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