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

pass shuffle flag in the constructor

Browse Source
This commit is contained in:
Tomasz Sobczyk
2020-10-23 22:19:50 +02:00
committed by nodchip
parent 31f94a18b3
commit 8fb208598b
2 changed files with 5 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/learn/learn.cpp
View File

@@ -383,11 +383,12 @@ namespace Learner
LearnerThink(
const std::vector<std::string>& filenames,
bool shuffle,
uint64_t thread_num,
const std::string& seed
) :
prng(seed),
sr(filenames, 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())),
learn_loss_sum{}
{
save_only_once = false;
@@ -403,11 +404,6 @@ namespace Learner
total_done = 0;
}
void set_do_shuffle(bool v)
{
sr.set_do_shuffle(v);
}
void learn(uint64_t epochs);
@@ -1150,7 +1146,7 @@ namespace Learner
Eval::NNUE::set_batch_size(nn_batch_size);
Eval::NNUE::set_options(nn_options);
LearnerThink learn_think(filenames, thread_num, seed);
LearnerThink learn_think(filenames, !no_shuffle, thread_num, seed);
if (newbob_decay != 1.0 && !Options["SkipLoadingEval"]) {
// Save the current net to [EvalSaveDir]\original.
@@ -1165,7 +1161,6 @@ namespace Learner
// Reflect other option settings.
learn_think.eval_limit = eval_limit;
learn_think.save_only_once = save_only_once;
learn_think.set_do_shuffle(!no_shuffle);
learn_think.reduction_gameply = reduction_gameply;
learn_think.newbob_decay = newbob_decay;