From df05ecb1d540b0a56936ab4960404da5f6e0b5d6 Mon Sep 17 00:00:00 2001 From: nodchip Date: Fri, 10 Jul 2020 16:14:19 +0900 Subject: [PATCH] Added halfkp_384x2-32-32. --- .../nnue/architectures/halfkp_384x2-32-32.h | 39 +++++++++++++++++++ src/eval/nnue/nnue_architecture.h | 1 + 2 files changed, 40 insertions(+) create mode 100644 src/eval/nnue/architectures/halfkp_384x2-32-32.h diff --git a/src/eval/nnue/architectures/halfkp_384x2-32-32.h b/src/eval/nnue/architectures/halfkp_384x2-32-32.h new file mode 100644 index 00000000..3d28139a --- /dev/null +++ b/src/eval/nnue/architectures/halfkp_384x2-32-32.h @@ -0,0 +1,39 @@ +// Definition of input features and network structure used in NNUE evaluation function + +#ifndef HALFKP_384X2_32_32_H +#define HALFKP_384X2_32_32_H + +#include "../features/feature_set.h" +#include "../features/half_kp.h" + +#include "../layers/input_slice.h" +#include "../layers/affine_transform.h" +#include "../layers/clipped_relu.h" + +namespace Eval { + +namespace NNUE { + +// Input features used in evaluation function +using RawFeatures = Features::FeatureSet< + Features::HalfKP>; + +// Number of input feature dimensions after conversion +constexpr IndexType kTransformedFeatureDimensions = 384; + +namespace Layers { + +// define network structure +using InputLayer = InputSlice; +using HiddenLayer1 = ClippedReLU>; +using HiddenLayer2 = ClippedReLU>; +using OutputLayer = AffineTransform; + +} // namespace Layers + +using Network = Layers::OutputLayer; + +} // namespace NNUE + +} // namespace Eval +#endif // HALFKP_384X2_32_32_H diff --git a/src/eval/nnue/nnue_architecture.h b/src/eval/nnue/nnue_architecture.h index 977c67fc..cb53e4f9 100644 --- a/src/eval/nnue/nnue_architecture.h +++ b/src/eval/nnue/nnue_architecture.h @@ -11,6 +11,7 @@ //#include "architectures/k-p-cr-ep_256x2-32-32.h" #include "architectures/halfkp_256x2-32-32.h" //#include "architectures/halfkp-cr-ep_256x2-32-32.h" +//#include "architectures/halfkp_384x2-32-32.h" namespace Eval {