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() {
AIDGE_THROW_OR_ABORT(std::runtime_error,
"GlobalAveragePooling : The input was not connected");
} else if (getInput(0)->dims().size() < 3) {
AIDGE_THROW_OR_ABORT(
std::runtime_error,
printf(
"GlobalAveragePooling : needs at least 3 dimensions input, "
"number of input dim : %lu",
"number of input dim : %lu\n",
getInput(0)->dims().size());
throw std::runtime_error("Global Average Pooling: Invalid input dim nb.");
} else {
// Global average pooling takes each filter, averages its values and uses it
// 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