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

fix : forgot to invert condition in if=>assert refactor

parent 045a3a4c
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!91Feat/operator global average pooling
......@@ -28,8 +28,8 @@ void Aidge::GlobalAveragePooling_Op::computeOutputDims() {
return;
// computation
} else {
AIDGE_ASSERT(getInput(0)->dims().size() < 3,
"GlobalAveragePooling : needs at least 3 dimensions input, "
AIDGE_ASSERT(getInput(0)->dims().size() >= 3,
"GlobalAveragePooling : needs at least a 3 dimensions input, "
"number of input dim : {}",
getInput(0)->dims().size());
// Global average pooling takes each filter, averages its values and uses
......
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