cudnn's exponetialAverageFactor is 1 - momentum

starimpact · Apr 25, 2016 · 89d81ad · 89d81ad
1 parent c27f406
commit 89d81ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/cudnn_batch_norm-inl.h b/src/operator/cudnn_batch_norm-inl.h
@@ -115,7 +115,7 @@ class CuDNNBatchNormOp : public Operator {
                                                       mean_desc_,
                                                       gamma.dptr_,
                                                       beta.dptr_,
-                                                      param_.momentum,
+                                                      1 - param_.momentum,
                                                       moving_mean.dptr_,
                                                       moving_inv_var.dptr_,
                                                       param_.eps,