Merge pull request #210 from dhbloo/master

Fix the incorrect NUM_PT in HalfKAv2_hmFactorized
official-stockfish · Oct 12, 2022 · 3532d8d · 3532d8d
2 parents 9fb6d8e + 0ea1c69
commit 3532d8d
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;