Fix the incorrect NUM_PT in HalfKAv2_hmFactorized

official-stockfish · Oct 12, 2022 · 0ea1c69 · 0ea1c69
1 parent 2072820
commit 0ea1c69
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/training_data_loader.cpp b/training_data_loader.cpp
@@ -318,7 +318,8 @@ struct HalfKAv2_hm {
 
 struct HalfKAv2_hmFactorized {
     // Factorized features
-    static constexpr int PIECE_INPUTS = HalfKAv2_hm::NUM_SQ * HalfKAv2_hm::NUM_PT;
+    static constexpr int NUM_PT = 12;
+    static constexpr int PIECE_INPUTS = HalfKAv2_hm::NUM_SQ * NUM_PT;
     static constexpr int INPUTS = HalfKAv2_hm::INPUTS + PIECE_INPUTS;
 
     static constexpr int MAX_PIECE_FEATURES = 32;