HCha/Stockfish
1
0
Fork 0
mirror of https://github.com/HChaZZY/Stockfish.git synced 2025-12-25 11:36:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow changing sfen reader buffer sizes for the learn command.

Browse Source
This commit is contained in:
Tomasz Sobczyk
2020-10-24 14:51:28 +02:00
committed by nodchip
parent d31169bab5
commit 371acaa0b5
3 changed files with 45 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/learn/learn.cpp
View File

@@ -385,10 +385,19 @@ namespace Learner
const std::vector<std::string>& filenames,
bool shuffle,
uint64_t thread_num,
const std::string& seed
const std::string& seed,
size_t read_size,
size_t buffer_size
) :
prng(seed),
sr(filenames, shuffle, SfenReaderMode::Cyclic, thread_num, std::to_string(prng.next_random_seed())),
sr(
filenames,
shuffle,
SfenReaderMode::Cyclic,
thread_num,
std::to_string(prng.next_random_seed()),
read_size,
buffer_size),
learn_loss_sum{}
{
save_only_once = false;
@@ -958,6 +967,9 @@ namespace Learner
uint64_t eval_save_interval = LEARN_EVAL_SAVE_INTERVAL;
uint64_t loss_output_interval = 1'000'000;
size_t sfen_read_size = SfenReader::DEFAULT_SFEN_READ_SIZE;
size_t thread_buffer_size = SfenReader::DEFAULT_THREAD_BUFFER_SIZE;
string validation_set_file_name;
string seed;
@@ -1045,6 +1057,9 @@ namespace Learner
else if (option == "dest_score_min_value") is >> dest_score_min_value;
else if (option == "dest_score_max_value") is >> dest_score_max_value;
else if (option == "sfen_read_size") is >> sfen_read_size;
else if (option == "thread_buffer_size") is >> thread_buffer_size;
else if (option == "seed") is >> seed;
else if (option == "set_recommended_uci_options")
{
@@ -1146,7 +1161,13 @@ namespace Learner
Eval::NNUE::set_batch_size(nn_batch_size);
Eval::NNUE::set_options(nn_options);
LearnerThink learn_think(filenames, !no_shuffle, thread_num, seed);
LearnerThink learn_think(
filenames,
!no_shuffle,
thread_num,
seed,
sfen_read_size,
thread_buffer_size);
if (newbob_decay != 1.0 && !Options["SkipLoadingEval"]) {
// Save the current net to [EvalSaveDir]\original.