Load modify_nnue from nnue-pytorch branch

linrock · Jul 3, 2024 · a8d191e · a8d191e
1 parent 4b8ede3
commit a8d191e
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/worker/games.py b/worker/games.py
@@ -25,6 +25,11 @@
 
 import requests
 
+# no-gpu-modify-nnue branch
+sys.path.insert(0, "../nnue-pytorch")
+from modify_nnue import modify_nnue
+
+
 IS_WINDOWS = "windows" in platform.system().lower()
 IS_MACOS = "darwin" in platform.system().lower()
 LOGFILE = "api.log"
@@ -1119,9 +1124,10 @@ def generate_tune_options(params):
         w_params = []
         b_params = []
 
-    # subprocess.check_output(["python3", "../../nnue-pytorch/load_nnue_test.py"])
-    # subprocess.check_output(["python3", "../../nnue-pytorch/modify_nnue.py", "nn-ddcfb9224cdb.nnue", "w_tune_options.csv"])
-    # subprocess.check_output(["python3", "../../nnue-pytorch/modify_nnue.py", "nn-ddcfb9224cdb.nnue", "b_tune_options.csv"])
+    # subprocess.check_output(["python3", "../nnue-pytorch/load_nnue_test.py"])
+    # subprocess.check_output(["python3", "../nnue-pytorch/modify_nnue.py", "nn-ddcfb9224cdb.nnue", "w_tune_options.csv"])
+    # subprocess.check_output(["python3", "../nnue-pytorch/modify_nnue.py", "nn-ddcfb9224cdb.nnue", "b_tune_options.csv"])
+    modify_nnue
 
     # Run cutechess-cli binary.
     # Stochastic rounding and probability for float N.p: (N, 1-p); (N+1, p)