X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Fnnue%2Fnnue_feature_transformer.h;h=a4a8e98f9c5e8f579cea140b77126f9763184421;hb=58054fd0fa6294510fc8cf76b0ba9673d5094c10;hp=f441274915329f3605831414841920a65093f0db;hpb=b748b46714d5f8e0acca0a042ede1fc95e4f5190;p=stockfish

diff --git a/src/nnue/nnue_feature_transformer.h b/src/nnue/nnue_feature_transformer.h
index f4412749..a4a8e98f 100644
--- a/src/nnue/nnue_feature_transformer.h
+++ b/src/nnue/nnue_feature_transformer.h
@@ -118,6 +118,15 @@ namespace Stockfish::Eval::NNUE {
       return !stream.fail();
     }
 
+    // Write network parameters
+    bool write_parameters(std::ostream& stream) const {
+      for (std::size_t i = 0; i < HalfDimensions; ++i)
+        write_little_endian<BiasType>(stream, biases[i]);
+      for (std::size_t i = 0; i < HalfDimensions * InputDimensions; ++i)
+        write_little_endian<WeightType>(stream, weights[i]);
+      return !stream.fail();
+    }
+
     // Convert input features
     void transform(const Position& pos, OutputType* output) const {
       update_accumulator(pos, WHITE);