Exporting the currently loaded network file

[stockfish] / src / nnue / nnue_feature_transformer.h

diff --git a/src/nnue/nnue_feature_transformer.h b/src/nnue/nnue_feature_transformer.h
index f441274915329f3605831414841920a65093f0db..a4a8e98f9c5e8f579cea140b77126f9763184421 100644 (file)
--- 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);