diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-12-03 02:56:52 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-12-03 02:56:52 +0300 |
commit | 1bcbf547a95ee013c563cd7bd7e04dc85b17fe4f (patch) | |
tree | 8308ae38abd7343c8dc037a3598661d94f58b4ca | |
parent | 72b57172ccbd791bf1b57a0ae1bce85fe447245c (diff) | |
parent | a62ee96646121ccd9e1e30ade786e823f275f889 (diff) |
Merge pull request #1059 from gchanan/extraShapeCheck
Remove extra size check in SpatialAveragePooling.
-rw-r--r-- | lib/THNN/generic/SpatialAveragePooling.c | 5 |
1 files changed, 0 insertions, 5 deletions
diff --git a/lib/THNN/generic/SpatialAveragePooling.c b/lib/THNN/generic/SpatialAveragePooling.c index fda1abd..60a5eb1 100644 --- a/lib/THNN/generic/SpatialAveragePooling.c +++ b/lib/THNN/generic/SpatialAveragePooling.c @@ -133,11 +133,6 @@ void THNN_(SpatialAveragePooling_updateOutput)( --outputWidth; } - THArgCheck(inputWidth >= kW - 2 * padW && inputHeight >= kH - 2 * padH, 2, - "input image smaller than (kernel size - 2 * padW). Got " - "inputHeight: %d inputWidth: %d kH %d kW %d padH %d padW %d", - inputHeight, inputWidth, kH, kW, padH, padW); - if (input->nDimension == 3) THTensor_(resize3d)(output, nInputPlane, outputHeight, outputWidth); else |