Fixed sign error in GPU implementation of ComplexSiglog

pvirtue · May 13, 2017 · 8b0763b · 8b0763b
1 parent ecda77c
commit 8b0763b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/layers/complex_siglog_layer.cu b/src/caffe/layers/complex_siglog_layer.cu
@@ -53,7 +53,7 @@ __global__ void ComplexSiglogBackward(const int n, const cuComplex* bottom,
     // Useful temp variable
     float c_r_sz_d = c + 1.0/r * sz_to_d;
 
-    float dfdz_numer = (c_r_sz_d)*s + s*(d/(2*r) * s_to_d * z_to_d);
+    float dfdz_numer = (c_r_sz_d)*s - s*(d/(2*r) * s_to_d * z_to_d);
     float dfdz_denom = c_r_sz_d * c_r_sz_d + 1e-14;
     cuComplex dfdz = make_cuFloatComplex(dfdz_numer/dfdz_denom, 0);