mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 17:54:27 +00:00
Add model summary
This commit is contained in:
@@ -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;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user