diff --git a/src/operator/GlobalAveragePooling.cpp b/src/operator/GlobalAveragePooling.cpp index 067d25077d967ef28de3ed1a2d4653b5101b37d1..da760a4c89203c0415bb9a0259e25d5e7908b7d6 100644 --- a/src/operator/GlobalAveragePooling.cpp +++ b/src/operator/GlobalAveragePooling.cpp @@ -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