From 979d02126b1a597361934f86f50eeda31ff083fe Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 09 Feb 2015 21:27:58 +0000
Subject: [PATCH] Generalizing conv layer so deconv is easier
---
src/darknet.c | 22 +---------------------
1 files changed, 1 insertions(+), 21 deletions(-)
diff --git a/src/darknet.c b/src/darknet.c
index 8bb5a74..0b93aa6 100644
--- a/src/darknet.c
+++ b/src/darknet.c
@@ -225,8 +225,7 @@
void train_imagenet(char *cfgfile, char *weightfile)
{
float avg_loss = -1;
- // TODO
- srand(0);
+ srand(time(0));
char *base = basename(cfgfile);
printf("%s\n", base);
network net = parse_network_cfg(cfgfile);
@@ -585,25 +584,6 @@
cvWaitKey(0);
}
-#ifdef GPU
-void test_convolutional_layer()
-{
- network net = parse_network_cfg("cfg/nist_conv.cfg");
- int size = get_network_input_size(net);
- float *in = calloc(size, sizeof(float));
- int i;
- for(i = 0; i < size; ++i) in[i] = rand_normal();
- convolutional_layer layer = *(convolutional_layer *)net.layers[0];
- int out_size = convolutional_out_height(layer)*convolutional_out_width(layer)*layer.batch;
- cuda_compare(layer.output_gpu, layer.output, out_size, "nothing");
- cuda_compare(layer.biases_gpu, layer.biases, layer.n, "biases");
- cuda_compare(layer.filters_gpu, layer.filters, layer.n*layer.size*layer.size*layer.c, "filters");
- bias_output(layer);
- bias_output_gpu(layer);
- cuda_compare(layer.output_gpu, layer.output, out_size, "biased output");
-}
-#endif
-
void test_correct_nist()
{
network net = parse_network_cfg("cfg/nist_conv.cfg");
--
Gitblit v1.10.0