Skip updating value if the node has become terminal.

LeelaChessZero · Mar 29, 2019 · 5f9c77e · 5f9c77e
1 parent 46bb54b
commit 5f9c77e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/mcts/node.cc b/src/mcts/node.cc
@@ -242,9 +242,11 @@ void Node::CancelScoreUpdate(int multivisit) {
 }
 
 void Node::FinalizeScoreUpdate(float v, float d, int multivisit) {
-  // Recompute Q.
-  q_ += multivisit * (v - q_) / (n_ + multivisit);
-  d_ += multivisit * (d - d_) / (n_ + multivisit);
+  // Skip updating value if the node has become terminal.
+  if (!is_terminal_) {
+    q_ += multivisit * (v - q_) / (n_ + multivisit);
+    d_ += multivisit * (d - d_) / (n_ + multivisit);
+  }
 
   // If first visit, update parent's sum of policies visited at least once.
   if (n_ == 0 && parent_ != nullptr) {