Allow configuring default komi at compile-time.

Pull request leela-zero#2257.
AncalagonX · Mar 19, 2019 · 6fb5b6b · 6fb5b6b
1 parent 8c5068a
commit 6fb5b6b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/src/GTP.cpp b/src/GTP.cpp
@@ -589,7 +589,7 @@ void GTP::execute(GameState & game, const std::string& xinput) {
     } else if (command.find("komi") == 0) {
         std::istringstream cmdstream(command);
         std::string tmp;
-        float komi;
+        float komi = KOMI;
         float old_komi = game.get_komi();
 
         cmdstream >> tmp;  // eat komi
@@ -635,7 +635,7 @@ void GTP::execute(GameState & game, const std::string& xinput) {
     } else if (command.find("narrow_search") == 0) {
 		//m_search_width = (1.11 * m_search_width); // Not used. Instead, this is performed with the below command in UCTNode.cpp.
 		UCTNode::narrow_search();
-		game.set_komi((cfg_manual_komi / 10.0f));
+		//game.set_komi((cfg_manual_komi / 10.0f)); // Shouldn't need this anymore since I have this command spammed everywhere else already, like in "genmove" below.
         return;
 
     } else if (command.find("genmove") == 0

diff --git a/src/Leela.cpp b/src/Leela.cpp
@@ -554,7 +554,7 @@ int main(int argc, char *argv[]) {
     auto maingame = std::make_unique<GameState>();
 
     /* set board limits */
-    maingame->init_game(BOARD_SIZE, TRAINED_UNIT_KOMI);
+    maingame->init_game(BOARD_SIZE, KOMI);
 
     if (cfg_benchmark) {
         cfg_quiet = false;

diff --git a/src/config.h b/src/config.h
@@ -53,10 +53,9 @@ static constexpr auto NUM_INTERSECTIONS = BOARD_SIZE * BOARD_SIZE;
 static constexpr auto POTENTIAL_MOVES = NUM_INTERSECTIONS + 1; // including pass
 
 /*
- * TRAINED_UNIT_KOMI: Define the komi used during training that represents a
- * value of 1.0 for the side to move NN inputs
+ * KOMI: Define the default komi to use when training.
  */
-static constexpr auto TRAINED_UNIT_KOMI = 7.5f;
+static constexpr auto KOMI = 7.5f;
 
 /*
  * Features