diff options
author | ningqingqun <ningqingqun@gmail.com> | 2017-05-24 16:05:08 +0300 |
---|---|---|
committer | ningqingqun <ningqingqun@gmail.com> | 2017-05-24 16:05:08 +0300 |
commit | a55fb7342224f8663197a5e2fcc65489583f5b3c (patch) | |
tree | 4c9e28d8d39005dda09703f7651d6c5d928d9aa2 | |
parent | f2b2286cae74689d83beb98ffa654f1f3cdefc06 (diff) |
Fix the mix-up of height and width on depth-wise convolution
-rw-r--r-- | lib/THCUNN/generic/SpatialDepthWiseConvolution.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THCUNN/generic/SpatialDepthWiseConvolution.cu b/lib/THCUNN/generic/SpatialDepthWiseConvolution.cu index 2520ff4..5da6570 100644 --- a/lib/THCUNN/generic/SpatialDepthWiseConvolution.cu +++ b/lib/THCUNN/generic/SpatialDepthWiseConvolution.cu @@ -232,7 +232,7 @@ void THNN_(SpatialDepthWiseConvolution_updateOutput)( THCTensor_(free)(state, _bias); // Transpose output - THCTensor_(resize4d)(state, output, batchSize, nInputPlane * nOutputPlane, outputWidth, outputHeight); + THCTensor_(resize4d)(state, output, batchSize, nInputPlane * nOutputPlane, outputHeight, outputWidth); // Make a contiguous copy of output (OPTIONAL) // THCTensor *_output = THCTensor_(newContiguous)(state, output); |