diff options
| author | Soumith Chintala <soumith@gmail.com> | 2017-06-07 06:31:54 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-06-07 06:31:54 +0300 |
| commit | 3087b8786be61c04270330b8fc2c628690ee0d0c (patch) | |
| tree | 2fac1f239c2f036256665d5328acb76d9a90c1e3 | |
| parent | 8e3364c64aa85e7ab4970c8d5f6a4f18d0c3eeae (diff) | |
| parent | a55fb7342224f8663197a5e2fcc65489583f5b3c (diff) | |
Merge pull request #470 from qqning/master
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); |