From f0aceda809a00a34df32a107090181ad91c8ee32 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Sun, 3 Apr 2022 14:27:38 -0400
Subject: [PATCH] Fix serialization for base layer.

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

diff --git a/src/mlpack/methods/ann/layer/layer.hpp b/src/mlpack/methods/ann/layer/layer.hpp
index 15bffea1d6..675232e6e2 100644
--- a/src/mlpack/methods/ann/layer/layer.hpp
+++ b/src/mlpack/methods/ann/layer/layer.hpp
@@ -277,9 +277,10 @@ class Layer
   template<typename Archive>
   void serialize(Archive& ar, const uint32_t /* version */)
   {
-    ar(inputDimensions);
-    ar(outputDimensions);
-    ar(training);
+    ar(CEREAL_NVP(inputDimensions));
+    ar(CEREAL_NVP(outputDimensions));
+    ar(CEREAL_NVP(validOutputDimensions));
+    ar(CEREAL_NVP(training));
 
     // Note that layer weights are serialized by the FFN!
   }