]> git.sesse.net Git - stockfish/blobdiff - src/nnue/evaluate_nnue.cpp
Embed default net, and simplify using non-default nets
[stockfish] / src / nnue / evaluate_nnue.cpp
index e66190892ad4029535bdd86c66ee20eeb6fc7f6d..d6ac9894cbbd4203303478c554a0cc5c1ce1ba78 100644 (file)
@@ -18,7 +18,6 @@
 
 // Code for calculating NNUE evaluation function
 
-#include <fstream>
 #include <iostream>
 #include <set>
 
@@ -143,17 +142,12 @@ namespace Eval::NNUE {
     return accumulator.score;
   }
 
-  // Load the evaluation function file
-  bool load_eval_file(const std::string& evalFile) {
+  // Load eval, from a file stream or a memory stream
+  bool load_eval(std::string streamName, std::istream& stream) {
 
     Initialize();
-    fileName = evalFile;
-
-    std::ifstream stream(evalFile, std::ios::binary);
-
-    const bool result = ReadParameters(stream);
-
-    return result;
+    fileName = streamName;
+    return ReadParameters(stream);
   }
 
   // Evaluation function. Perform differential calculation.