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

Cleaner and more outputs during training initialization.

Browse Source
This commit is contained in:
Tomasz Sobczyk
2020-10-24 14:21:59 +02:00
committed by nodchip
parent b882423005
commit 2c477d76ec
12 changed files with 129 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/nnue/evaluate_nnue_learner.cpp
View File

@@ -54,23 +54,28 @@ namespace Eval::NNUE {
} // namespace
// Initialize learning
void initialize_training(const std::string& seed) {
std::cout << "Initializing NN training for "
<< get_architecture_string() << std::endl;
void initialize_training(
const std::string& seed,
SynchronizedRegionLogger::Region& out) {
std::cout << std::endl;
out << "INFO (initialize_training): Initializing NN training for "
<< get_architecture_string() << std::endl;
std::cout << "Layers:\n"
<< get_layers_info() << std::endl;
out << std::endl;
std::cout << std::endl;
out << "Layers:\n"
<< get_layers_info() << std::endl;
out << std::endl;
assert(feature_transformer);
assert(network);
trainer = Trainer<Network>::create(network.get(), feature_transformer.get());
rng.seed(PRNG(seed).rand<uint64_t>());
if (Options["SkipLoadingEval"]) {
out << "INFO (initialize_training): Performing random net initialization.\n";
trainer->initialize(rng);
}
}