diff --git a/src/nnue/network.cpp b/src/nnue/network.cpp index b37fc272f..f50fd5262 100644 --- a/src/nnue/network.cpp +++ b/src/nnue/network.cpp @@ -253,8 +253,12 @@ void Network::verify(std::string evalfilePath) const { exit(EXIT_FAILURE); } + size_t size = sizeof(*featureTransformer) + sizeof(*network) * LayerStacks; if (Cluster::is_root()) - sync_cout << "info string NNUE evaluation using " << evalfilePath << sync_endl; + sync_cout << "info string NNUE evaluation using " << evalfilePath << " (" + << size / (1024 * 1024) << "MiB, (" << featureTransformer->InputDimensions << ", " + << network[0]->TransformedFeatureDimensions << ", " << network[0]->FC_0_OUTPUTS + << ", " << network[0]->FC_1_OUTPUTS << ", 1))" << sync_endl; }