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
---
cfg/darknet.cfg | 9 +++++----
src/convolutional_kernels.cu | 6 ++++++
2 files changed, 11 insertions(+), 4 deletions(-)
diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg
index 85497a3..00e9c36 100644
--- a/cfg/darknet.cfg
+++ b/cfg/darknet.cfg
@@ -7,10 +7,11 @@
momentum=0.9
decay=0.0005
-learning_rate=0.1
-policy=poly
-power=4
-max_batches=500000
+learning_rate=0.01
+policy=sigmoid
+gamma=.00002
+step=400000
+max_batches=800000
[crop]
crop_height=224
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