apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixing minor bug

Browse Source
This commit is contained in:
shubham1206agra
2022-05-28 22:05:14 +05:30
parent 49d3dd4896
commit 6b9dc533d9
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/adaptive_max_pooling_impl.hpp
+2 -2
View File
@@ -114,8 +114,8 @@ void AdaptiveMaxPoolingType<MatType>::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 ("
src/mlpack/methods/ann/layer/adaptive_mean_pooling_impl.hpp
+2 -2
View File
@@ -114,8 +114,8 @@ void AdaptiveMeanPoolingType<MatType>::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 ("