chore: hopefully fix policy inference

raklaptudirm · Jun 6, 2024 · dd9b3d6 · dd9b3d6
1 parent d7f6efb
commit dd9b3d6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/mcts/policy.rs b/src/mcts/policy.rs
@@ -50,7 +50,12 @@ impl PolicyNetwork {
         let from_subnet = &self.subnets[(mov.source() as usize).min(49)];
         let from_vec = from_subnet.out(feats);
 
-        let to_subnet = &self.subnets[50 + (mov.target() as usize).min(48)];
+        let target = if mov.is_single() {
+            63
+        } else {
+            mov.target() as usize
+        };
+        let to_subnet = &self.subnets[50 + (target).min(48)];
         let to_vec = to_subnet.out(feats);
 
         from_vec.dot(&to_vec)