opelly/Stockfish
1
0
Fork 0
mirror of https://github.com/opelly27/Stockfish.git synced 2026-05-20 15:37:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

add option ensure_quiet for gensfen that makes the generated position quiet

Browse Source
This commit is contained in:
Tomasz Sobczyk
2020-11-09 19:21:27 +01:00
committed by nodchip
parent 9b930023fb
commit 00797a3d86
1 changed files with 86 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/learn/gensfen.cpp
+86 -7
View File
@@ -94,6 +94,8 @@ namespace Learner
bool detect_draw_by_consecutive_low_score = true; bool detect_draw_by_consecutive_low_score = true;
bool detect_draw_by_insufficient_mating_material = true; bool detect_draw_by_insufficient_mating_material = true;
bool ensure_quiet = false;
uint64_t num_threads; uint64_t num_threads;
void enforce_constraints() void enforce_constraints()
@@ -316,19 +318,86 @@ namespace Learner
// Discard stuff before write_minply is reached // Discard stuff before write_minply is reached
// because it can harm training due to overfitting. // because it can harm training due to overfitting.
// Initial positions would be too common. // Initial positions would be too common.
if (ply >= params.write_minply && !was_seen_before(pos)) if (ply >= params.write_minply)
{ {
packed_sfens.emplace_back(PackedSfenValue()); packed_sfens.emplace_back(PackedSfenValue());
auto& psv = packed_sfens.back(); auto& psv = packed_sfens.back();
// Here we only write the position data. if (params.ensure_quiet)
// Result is added after the whole game is done. {
pos.sfen_pack(psv.sfen); auto [qsearch_value, qsearch_pv] = Search::qsearch(pos);
if (qsearch_pv.empty())
{
// Here we only write the position data.
// Result is added after the whole game is done.
pos.sfen_pack(psv.sfen);
psv.score = search_value; // Already a quiet position
psv.gamePly = ply; psv.score = search_value;
psv.move = search_pv[0]; psv.move = search_pv[0];
psv.gamePly = ply;
}
else
{
// Navigate to a quiet
int old_ply = ply;
for (auto m : qsearch_pv)
{
pos.do_move(m, states[ply++]);
}
if (was_seen_before(pos))
{
// Just skip the move.
packed_sfens.pop_back();
}
else
{
// Reevaluate
auto [quiet_search_value, quiet_search_pv] = Search::search(pos, depth, 1, params.nodes);
if (quiet_search_pv.empty())
{
// Just skip the move.
packed_sfens.pop_back();
}
else
{
// Here we only write the position data.
// Result is added after the whole game is done.
pos.sfen_pack(psv.sfen);
psv.score = quiet_search_value;
psv.move = quiet_search_pv[0];
psv.gamePly = ply;
}
}
// Get back to the game
for (auto it = qsearch_pv.rbegin(); it != qsearch_pv.rend(); ++it)
{
pos.undo_move(*it);
}
ply = old_ply;
}
}
else
{
if (was_seen_before(pos))
{
packed_sfens.pop_back();
}
else
{
// Here we only write the position data.
// Result is added after the whole game is done.
pos.sfen_pack(psv.sfen);
psv.score = search_value;
psv.move = search_pv[0];
psv.gamePly = ply;
}
}
} }
// Update the next move according to best search result or random move. // Update the next move according to best search result or random move.
@@ -777,6 +846,10 @@ namespace Learner
UCI::setoption("PruneAtShallowDepth", "false"); UCI::setoption("PruneAtShallowDepth", "false");
UCI::setoption("EnableTranspositionTable", "true"); UCI::setoption("EnableTranspositionTable", "true");
} }
else if (token == "ensure_quiet")
{
params.ensure_quiet = true;
}
else else
cout << "ERROR: Ignoring unknown option " << token << endl; cout << "ERROR: Ignoring unknown option " << token << endl;
} }
@@ -791,6 +864,12 @@ namespace Learner
cout << "WARNING: Unknown sfen format `" << sfen_format << "`. Using bin\n"; cout << "WARNING: Unknown sfen format `" << sfen_format << "`. Using bin\n";
} }
if (params.ensure_quiet)
{
// Otherwise we can't ensure quiet positions...
UCI::setoption("EnableTranspositionTable", "false");
}
if (random_file_name) if (random_file_name)
{ {
// Give a random number to output_file_name at this point. // Give a random number to output_file_name at this point.