From 7c120aef23fde5b215b0fb6eef3074a15f16ff69 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 19 Nov 2014 22:03:51 +0000
Subject: [PATCH] stable, dropout on gpu
---
src/cnn.c | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/cnn.c b/src/cnn.c
index 5399679..f77b8f0 100644
--- a/src/cnn.c
+++ b/src/cnn.c
@@ -308,8 +308,8 @@
void train_imagenet()
{
float avg_loss = 1;
- network net = parse_network_cfg("/home/pjreddie/imagenet_backup/imagenet_2280.cfg");
- //network net = parse_network_cfg("cfg/imagenet2.cfg");
+ //network net = parse_network_cfg("/home/pjreddie/imagenet_backup/alexnet_1270.cfg");
+ network net = parse_network_cfg("cfg/imagenet.cfg");
printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
int imgs = 1000/net.batch+1;
srand(time(0));
@@ -1042,6 +1042,7 @@
#ifdef GPU
else if(0==strcmp(argv[1], "test_gpu")) test_gpu_blas();
#endif
+ test_parser();
fprintf(stderr, "Success!\n");
return 0;
}
--
Gitblit v1.10.0