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

Move setting learn search limits to learner.

Browse Source
This commit is contained in:
Tomasz Sobczyk
2020-10-24 23:17:56 +02:00
committed by nodchip
parent c229929d26
commit e4868cb59e
1 changed files with 27 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
src/learn/learn.cpp
View File

@@ -467,6 +467,8 @@ namespace Learner
void learn(uint64_t epochs);
private:
static void set_learning_search_limits();
void learn_worker(Thread& th, std::atomic<uint64_t>& counter, uint64_t limit);
void update_weights(const PSVector& psv, uint64_t epoch);
@@ -510,13 +512,37 @@ namespace Learner
AtomicLoss learn_loss_sum;
};
void LearnerThink::set_learning_search_limits()
{
Threads.main()->ponder = false;
// About Search::Limits
// Be careful because this member variable is global and affects other threads.
auto& limits = Search::Limits;
limits.startTime = now();
// Make the search equivalent to the "go infinite" command. (Because it is troublesome if time management is done)
limits.infinite = true;
// Since PV is an obstacle when displayed, erase it.
limits.silent = true;
// If you use this, it will be compared with the accumulated nodes of each thread. Therefore, do not use it.
limits.nodes = 0;
// depth is also processed by the one passed as an argument of Learner::search().
limits.depth = 0;
}
void LearnerThink::learn(uint64_t epochs)
{
#if defined(_OPENMP)
omp_set_num_threads((int)Options["Threads"]);
#endif
set_learning_search_limits();
Eval::NNUE::verify_any_net_loaded();
const PSVector sfen_for_mse =
@@ -929,27 +955,6 @@ namespace Learner
return false;
}
static void set_learning_search_limits()
{
// About Search::Limits
// Be careful because this member variable is global and affects other threads.
auto& limits = Search::Limits;
limits.startTime = now();
// Make the search equivalent to the "go infinite" command. (Because it is troublesome if time management is done)
limits.infinite = true;
// Since PV is an obstacle when displayed, erase it.
limits.silent = true;
// If you use this, it will be compared with the accumulated nodes of each thread. Therefore, do not use it.
limits.nodes = 0;
// depth is also processed by the one passed as an argument of Learner::search().
limits.depth = 0;
}
// Learning from the generated game record
void learn(istringstream& is)
{
@@ -1159,10 +1164,6 @@ namespace Learner
out << "INFO: Started initialization." << endl;
Threads.main()->ponder = false;
set_learning_search_limits();
Eval::NNUE::initialize_training(params.seed, out);
Eval::NNUE::set_batch_size(nn_batch_size);
Eval::NNUE::set_options(nn_options);