commit df1c890a10024ab414523b22f08148bd9b0c2723
author Adam Paszke <adam.paszke@gmail.com> Thu Aug 11 12:37:11 2016 -0700
committer Adam Paszke <adam.paszke@gmail.com> Thu Aug 11 12:37:11 2016 -0700
tree e7d4848f2133d2a04a684bed85ad36dd3ca5d83e
parent 959b12f1b8067906ae36c5c918ebf2f4f528f5ef

Fix DistKLDivCriterion gradInput formula

DistKLDivCriterion.cu

diff --git a/DistKLDivCriterion.cu b/DistKLDivCriterion.cu
index 93bc2e6..2984672 100644
--- a/DistKLDivCriterion.cu
+++ b/DistKLDivCriterion.cu
@@ -64,7 +64,7 @@
              "input and target need to have the same number of elements");
 
   long size = THCudaTensor_nElement(state, input);
-  float norm = (sizeAverage ? 2./size : 2.);
+  float norm = (sizeAverage ? 1./size : 1.);
 
   input = THCudaTensor_newContiguous(state, input);
   target = THCudaTensor_newContiguous(state, target);