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!