From 892923514f66d9b6acefd44f3ddf688bd7e6f268 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 08 Dec 2015 23:12:10 +0000
Subject: [PATCH] fixed darknet, stuff
---
src/convolutional_kernels.cu | 6 ++++++
1 files changed, 6 insertions(+), 0 deletions(-)
diff --git a/src/convolutional_kernels.cu b/src/convolutional_kernels.cu
index 130824a..a64a499 100644
--- a/src/convolutional_kernels.cu
+++ b/src/convolutional_kernels.cu
@@ -122,6 +122,12 @@
fast_mean_gpu(l.output_gpu, l.batch, l.n, l.out_h*l.out_w, l.mean_gpu);
fast_variance_gpu(l.output_gpu, l.mean_gpu, l.batch, l.n, l.out_h*l.out_w, l.variance_gpu);
+ /*
+ cuda_pull_array(l.variance_gpu, l.mean, 1);
+ printf("%f\n", l.mean[0]);
+ */
+
+
scal_ongpu(l.n, .95, l.rolling_mean_gpu, 1);
axpy_ongpu(l.n, .05, l.mean_gpu, 1, l.rolling_mean_gpu, 1);
scal_ongpu(l.n, .95, l.rolling_variance_gpu, 1);
--
Gitblit v1.10.0