Print layers and their indices during training initialization.

This commit is contained in:
Tomasz Sobczyk
2020-10-24 10:52:49 +02:00
committed by nodchip
parent 3bf397a569
commit be3937c37b
8 changed files with 152 additions and 51 deletions

View File

@@ -58,6 +58,13 @@ namespace Eval::NNUE {
std::cout << "Initializing NN training for "
<< get_architecture_string() << std::endl;
std::cout << std::endl;
std::cout << "Layers:\n"
<< get_layers_info() << std::endl;
std::cout << std::endl;
assert(feature_transformer);
assert(network);
trainer = Trainer<Network>::create(network.get(), feature_transformer.get());