mirror of
https://github.com/HChaZZY/Stockfish.git
synced 2025-12-25 03:26:24 +08:00
Added halfkp_384x2-32-32.
This commit is contained in:
39
src/eval/nnue/architectures/halfkp_384x2-32-32.h
Normal file
39
src/eval/nnue/architectures/halfkp_384x2-32-32.h
Normal file
@@ -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<Features::Side::kFriend>>;
|
||||
|
||||
// Number of input feature dimensions after conversion
|
||||
constexpr IndexType kTransformedFeatureDimensions = 384;
|
||||
|
||||
namespace Layers {
|
||||
|
||||
// define network structure
|
||||
using InputLayer = InputSlice<kTransformedFeatureDimensions * 2>;
|
||||
using HiddenLayer1 = ClippedReLU<AffineTransform<InputLayer, 32>>;
|
||||
using HiddenLayer2 = ClippedReLU<AffineTransform<HiddenLayer1, 32>>;
|
||||
using OutputLayer = AffineTransform<HiddenLayer2, 1>;
|
||||
|
||||
} // namespace Layers
|
||||
|
||||
using Network = Layers::OutputLayer;
|
||||
|
||||
} // namespace NNUE
|
||||
|
||||
} // namespace Eval
|
||||
#endif // HALFKP_384X2_32_32_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 {
|
||||
|
||||
|
||||
Reference in New Issue
Block a user