From 6b9dc533d92739f4e2d94d900d302de4643e5a28 Mon Sep 17 00:00:00 2001 From: shubham1206agra Date: Sat, 28 May 2022 22:05:14 +0530 Subject: [PATCH] fixing minor bug --- src/mlpack/methods/ann/layer/adaptive_max_pooling_impl.hpp | 4 ++-- src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/mlpack/methods/ann/layer/adaptive_max_pooling_impl.hpp b/src/mlpack/methods/ann/layer/adaptive_max_pooling_impl.hpp index 7be6a68929..8dead7fb29 100644 --- a/src/mlpack/methods/ann/layer/adaptive_max_pooling_impl.hpp +++ b/src/mlpack/methods/ann/layer/adaptive_max_pooling_impl.hpp @@ -114,8 +114,8 @@ void AdaptiveMaxPoolingType::ComputeOutputDimensions() this->outputDimensions[0] = outputWidth; this->outputDimensions[1] = outputHeight; - if (outputWidth < this->inputDimensions[0] || - outputHeight < this->inputDimensions[1]) + if (outputWidth > this->inputDimensions[0] || + outputHeight > this->inputDimensions[1]) { Log::Fatal << "Given output shape (" << outputWidth << ", " << outputHeight << ") is not possible for given input shape (" diff --git a/src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp b/src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp index dfb3a02807..cd6c2607a6 100644 --- a/src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp +++ b/src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp @@ -114,8 +114,8 @@ void AdaptiveMeanPoolingType::ComputeOutputDimensions() this->outputDimensions[0] = outputWidth; this->outputDimensions[1] = outputHeight; - if (outputWidth < this->inputDimensions[0] || - outputHeight < this->inputDimensions[1]) + if (outputWidth > this->inputDimensions[0] || + outputHeight > this->inputDimensions[1]) { Log::Fatal << "Given output shape (" << outputWidth << ", " << outputHeight << ") is not possible for given input shape ("