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 ("