diff options
| author | Gregory Chanan <gchanan@fb.com> | 2016-12-03 02:36:11 +0300 |
|---|---|---|
| committer | Gregory Chanan <gchanan@fb.com> | 2016-12-03 02:36:11 +0300 |
| commit | a62ee96646121ccd9e1e30ade786e823f275f889 (patch) | |
| tree | 8308ae38abd7343c8dc037a3598661d94f58b4ca | |
| parent | 72b57172ccbd791bf1b57a0ae1bce85fe447245c (diff) | |
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 |