mirror of
https://github.com/sockspls/badfish
synced 2025-04-30 16:53:09 +00:00
Added halfkp_384x2-32-32.
This commit is contained in:
parent
1de1eb2d0d
commit
df05ecb1d5
2 changed files with 40 additions and 0 deletions
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 {
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue