opelly/Stockfish
1
0
Fork 0
mirror of https://github.com/opelly27/Stockfish.git synced 2026-05-20 13:17:44 +00: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
src/nnue/evaluate_nnue_learner.cpp
+4 -1
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;