From c2f34e49a7493dc0bea629e049d8aebd8a98e296 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Wed, 13 Mar 2024 15:19:08 +0100
Subject: [PATCH] fix : error message
---
src/operator/GlobalAveragePooling.cpp | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/operator/GlobalAveragePooling.cpp b/src/operator/GlobalAveragePooling.cpp
index f2cfdc8bd..114c0d5f5 100644
--- a/src/operator/GlobalAveragePooling.cpp
+++ b/src/operator/GlobalAveragePooling.cpp
@@ -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).
--
GitLab