Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add model summary

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-28 18:42:58 +02:00
parent 6d6f4d4185
commit 098fd65074
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/model/model.hpp
View File

@@ -39,6 +39,8 @@ class Model {
bool validate(); bool validate();
void printSummary();
protected: protected:
Layers::Input* inputLayer; Layers::Input* inputLayer;
Layers::Output* outputLayer; Layers::Output* outputLayer;

8
src/model/model.cpp
View File

@@ -187,3 +187,11 @@ bool Model::validate() {
return valid; 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;
}
}