Fix serialization for base layer.

This commit is contained in:
Ryan Curtin
2022-04-03 14:27:38 -04:00
parent 2413a1cdf5
commit f0aceda809
+4 -3
View File
@@ -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!
}