diff --git a/Makefile b/Makefile index d1d77f6..c22f230 100644 --- a/Makefile +++ b/Makefile @@ -3,7 +3,7 @@ OBJ_DIR=build MOLY_DIR=src CXX=clang++ -DEFAULT_NET_NAME=mcabBaseline-40.bin +DEFAULT_NET_NAME=k64.nnue DEFAULT_EXE = $(OBJ_DIR)/Molybdenum DATAGEN_EXE = $(OBJ_DIR)/Datagen diff --git a/src/Nets/k64.nnue b/src/Nets/k64.nnue new file mode 100644 index 0000000..6b8c058 Binary files /dev/null and b/src/Nets/k64.nnue differ diff --git a/src/nnue.h b/src/nnue.h index 58b022b..c8cdc44 100644 --- a/src/nnue.h +++ b/src/nnue.h @@ -13,7 +13,7 @@ enum Toggle { }; static const int INPUT_SIZE = 12 * 64; -static const int L1_SIZE = 32; +static const int L1_SIZE = 64; static const int OUTPUT_SIZE = 1; static const int NET_SIZE = 3; static const std::array LAYER_SIZE = {INPUT_SIZE, L1_SIZE, OUTPUT_SIZE}; @@ -52,7 +52,7 @@ inline int screlu(int16_t input) { template inline int index(int pc, int sq) { - int square = C ? sq : sq ^ 56; + int square = C ? sq ^ 7 : sq ^ 63; int piece = C ? pc : makePiece(typeOf(pc), !colorOf(pc)); return piece * 64 + square;