From ee5df397740c6c606c4d6385c7b6971050943ac2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Mon, 18 Mar 2024 11:50:48 +0100
Subject: [PATCH] fix : forgot to invert condition in if=>assert refactor
---
src/operator/GlobalAveragePooling.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/operator/GlobalAveragePooling.cpp b/src/operator/GlobalAveragePooling.cpp
index 067d25077..da760a4c8 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
--
GitLab