From 6d69ea45f10f6724561d2dd2a20d58e8d32ba29a Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Sun, 3 Apr 2022 13:18:45 -0400
Subject: [PATCH] Clear member that is not serialized on loading.

---
 src/mlpack/methods/ann/layer/alpha_dropout_impl.hpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/mlpack/methods/ann/layer/alpha_dropout_impl.hpp b/src/mlpack/methods/ann/layer/alpha_dropout_impl.hpp
index 5b76b587ba..91a150f410 100644
--- a/src/mlpack/methods/ann/layer/alpha_dropout_impl.hpp
+++ b/src/mlpack/methods/ann/layer/alpha_dropout_impl.hpp
@@ -129,7 +129,9 @@ void AlphaDropoutType<MatType>::serialize(
   ar(CEREAL_NVP(b));
 
   // No need to serialize the mask, since it will be recomputed on the next
-  // forward pass.
+  // forward pass.  But we should clear it if we are loading.
+  if (Archive::is_loading::value)
+    mask.clear();
 }
 
 } // namespace ann