diff --git a/src/learn/learn.cpp b/src/learn/learn.cpp index cc51b04e..3e4f9495 100644 --- a/src/learn/learn.cpp +++ b/src/learn/learn.cpp @@ -502,8 +502,6 @@ namespace Learner << "Error reading sfen_for_mse. Read " << sfen_for_mse.size() << " out of " << sfen_for_mse_size << '\n'; - sr.stop(); - return; } @@ -541,8 +539,6 @@ namespace Learner break; } - sr.stop(); - Eval::NNUE::finalize_net(); save(true); diff --git a/src/learn/sfen_reader.h b/src/learn/sfen_reader.h index 78bf4ee8..0ef9765b 100644 --- a/src/learn/sfen_reader.h +++ b/src/learn/sfen_reader.h @@ -61,6 +61,8 @@ namespace Learner{ ~SfenReader() { + stop_flag = true; + if (file_worker_thread.joinable()) file_worker_thread.join(); } @@ -310,11 +312,6 @@ namespace Learner{ } } - void stop() - { - stop_flag = true; - } - void set_do_shuffle(bool v) { shuffle = v;