feat(nnue): new network

dogeystamp · Feb 10, 2025 · 69b196d · 69b196d
1 parent ef178a3
commit 69b196d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/nnue/s2_process_pgn_data.py b/nnue/s2_process_pgn_data.py
@@ -130,7 +130,7 @@ async def worker(game_generator: AsyncIterator[pgn.Game]) -> None:
     Puts rows of output into a global queue.
     """
     transport, engine = await chess.engine.popen_uci(args.engine)
-    await engine.configure(dict(NNUETrainInfo="true", Hash="128"))
+    await engine.configure(dict(NNUETrainInfo="true", Hash="800"))
 
     async for game in game_generator:
         wdl: int | None = None

diff --git a/nnue/s3_train_neural_net.py b/nnue/s3_train_neural_net.py
@@ -70,7 +70,7 @@
 BATCH_SIZE = 16384
 BIG_BATCH_SIZE = 2**22
 
-EPOCHS = 5
+EPOCHS = 10
 
 # neural net architecture
 

diff --git a/src/weights.bin b/src/weights.bin