Serialize output dimensions also.

This commit is contained in:
Ryan Curtin
2021-07-07 19:08:12 -04:00
parent a87672665c
commit d180cc3639
+1
View File
@@ -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!