From d180cc36393c1bfea4bd66d8a9df0efb05c8e082 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Wed, 7 Jul 2021 19:08:12 -0400
Subject: [PATCH] Serialize output dimensions also.

---
 src/mlpack/methods/ann/layer/layer.hpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/mlpack/methods/ann/layer/layer.hpp b/src/mlpack/methods/ann/layer/layer.hpp
index b2f520452e..2c63de5b86 100644
--- a/src/mlpack/methods/ann/layer/layer.hpp
+++ b/src/mlpack/methods/ann/layer/layer.hpp
@@ -255,6 +255,7 @@ class Layer
   void serialize(Archive& ar, const uint32_t /* version */)
   {
     ar(inputDimensions);
+    ar(outputDimensions);
     ar(training);
 
     // Note that layer weights are serialized by the FFN!