From 0ea1c69e4aad680af64c7d27fabd65b666927612 Mon Sep 17 00:00:00 2001 From: dblue Date: Wed, 12 Oct 2022 17:31:33 +0800 Subject: [PATCH] Fix the incorrect NUM_PT in HalfKAv2_hmFactorized --- training_data_loader.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/training_data_loader.cpp b/training_data_loader.cpp index 9aa81505..f2c33177 100644 --- 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;