diff --git a/include/model/model.hpp b/include/model/model.hpp index 815298d..9933e35 100644 --- a/include/model/model.hpp +++ b/include/model/model.hpp @@ -39,6 +39,8 @@ class Model { bool validate(); + void printSummary(); + protected: Layers::Input* inputLayer; Layers::Output* outputLayer; diff --git a/src/model/model.cpp b/src/model/model.cpp index 9319e1d..f493d22 100644 --- a/src/model/model.cpp +++ b/src/model/model.cpp @@ -186,4 +186,12 @@ bool Model::validate() { } return valid; +} + +void Model::printSummary() { + std::cout << "Model summary:" << std::endl; + for (const auto& layer : layers) { + std::cout << "Layer: " << layer.first << ", input size: " << layer.second->getInputSize() << ", output size: " + << layer.second->getOutputSize() << std::endl; + } } \ No newline at end of file