Skip to content
Snippets Groups Projects
Commit c2f34e49 authored by Grégoire Kubler's avatar Grégoire Kubler
Browse files

fix : error message

parent 5177fc0b
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!91Feat/operator global average pooling
...@@ -22,11 +22,11 @@ void Aidge::GlobalAveragePooling_Op::computeOutputDims() { ...@@ -22,11 +22,11 @@ void Aidge::GlobalAveragePooling_Op::computeOutputDims() {
AIDGE_THROW_OR_ABORT(std::runtime_error, AIDGE_THROW_OR_ABORT(std::runtime_error,
"GlobalAveragePooling : The input was not connected"); "GlobalAveragePooling : The input was not connected");
} else if (getInput(0)->dims().size() < 3) { } else if (getInput(0)->dims().size() < 3) {
AIDGE_THROW_OR_ABORT( printf(
std::runtime_error,
"GlobalAveragePooling : needs at least 3 dimensions input, " "GlobalAveragePooling : needs at least 3 dimensions input, "
"number of input dim : %lu", "number of input dim : %lu\n",
getInput(0)->dims().size()); getInput(0)->dims().size());
throw std::runtime_error("Global Average Pooling: Invalid input dim nb.");
} else { } else {
// Global average pooling takes each filter, averages its values and uses it // Global average pooling takes each filter, averages its values and uses it
// as an output(Much like a fancier flatten). // as an output(Much like a fancier flatten).
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment