Let's see if the serialization of the wrapped layer is the issue.

2019-11-24 14:13:44 +01:00
parent 41699414cb
commit 9c3f113bd5
1 changed files with 1 additions and 1 deletions
@@ -152,7 +152,7 @@ void WeightNorm<InputDataType, OutputDataType, CustomLayers...>::serialize(
    boost::apply_visitor(deleteVisitor, wrappedLayer);
  }

-  ar & BOOST_SERIALIZATION_NVP(wrappedLayer);
+  //ar & BOOST_SERIALIZATION_NVP(wrappedLayer);
  ar & BOOST_SERIALIZATION_NVP(layerWeightSize);

  // If we are loading, we need to initialize the weights.