From 0541c66b2ada044016c9c5fc704f16b2475d19ed Mon Sep 17 00:00:00 2001 From: Linmiao Xu Date: Wed, 3 Jul 2024 09:28:26 -0400 Subject: [PATCH] Use modified nnue instead of setting spsa params at runtime --- worker/games.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/worker/games.py b/worker/games.py index 428e345c2..8a3dedba2 100644 --- a/worker/games.py +++ b/worker/games.py @@ -1134,18 +1134,32 @@ def generate_tune_options(params): print(cmd) print() + # use modified nnue instead of setting spsa params at runtime + idx = cmd.index('option.EvalFile=nn-ddcfb9224cdb.nnue') + cmd = ( + cmd[:idx] + + f"option.EvalFile={w_spsa_nnue}" + + cmd[idx + 1 :] + ) + idx = cmd.index('option.EvalFile=nn-ddcfb9224cdb.nnue') + cmd = ( + cmd[:idx] + + f"option.EvalFile={b_spsa_nnue}" + + cmd[idx + 1 :] + ) + # Run cutechess-cli binary. # Stochastic rounding and probability for float N.p: (N, 1-p); (N+1, p) idx = cmd.index("_spsa_") cmd = ( cmd[:idx] - + (["option.TuneFile=w_tune_options.csv"] if len(w_params) != 0 else []) + # + (["option.TuneFile=w_tune_options.csv"] if len(w_params) != 0 else []) + cmd[idx + 1 :] ) idx = cmd.index("_spsa_") cmd = ( cmd[:idx] - + (["option.TuneFile=b_tune_options.csv"] if len(b_params) != 0 else []) + # + (["option.TuneFile=b_tune_options.csv"] if len(b_params) != 0 else []) + cmd[idx + 1 :] )