HCha/Stockfish
1
0
Fork 0
mirror of https://github.com/HChaZZY/Stockfish.git synced 2025-12-24 19:16:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Pass the new loss function to update_parameters

Browse Source
This commit is contained in:
Tomasz Sobczyk
2020-11-29 11:55:15 +01:00
committed by nodchip
parent 5a58eb803a
commit b71d1e8620
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/nnue/evaluate_nnue_learner.cpp
View File

@@ -195,8 +195,11 @@ namespace Eval::NNUE {
uint64_t epoch,
bool verbose,
double learning_rate,
Learner::CalcGradFunc calc_grad)
Learner::CalcGradFunc calc_grad,
Learner::CalcLossFunc calc_loss)
{
using namespace Learner::Autograd::UnivariateStatic;
assert(batch_size > 0);
learning_rate /= batch_size;

3
src/nnue/evaluate_nnue_learner.h
View File

@@ -38,7 +38,8 @@ namespace Eval::NNUE {
uint64_t epoch,
bool verbose,
double learning_rate,
Learner::CalcGradFunc calc_grad);
Learner::CalcGradFunc calc_grad,
Learner::CalcLossFunc calc_loss);
// Check if there are any problems with learning
void check_health();