From edbccdfcaf46f11e631afe98796f3e6e170da5d0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sun, 26 Oct 2014 05:04:34 +0000
Subject: [PATCH] Maybe something changed?
---
src/cnn.c | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/cnn.c b/src/cnn.c
index 1488c77..2d09582 100644
--- a/src/cnn.c
+++ b/src/cnn.c
@@ -499,7 +499,7 @@
int iters = 10000/net.batch;
while(++count <= 2000){
clock_t start = clock(), end;
- float loss = train_network_sgd(net, train, iters);
+ float loss = train_network_sgd_gpu(net, train, iters);
end = clock();
float test_acc = network_accuracy(net, test);
//float test_acc = 0;
@@ -957,8 +957,9 @@
int main(int argc, char *argv[])
{
- //test_gpu_blas();
- train_imagenet();
+ test_gpu_blas();
+ //train_imagenet();
+ //train_nist();
fprintf(stderr, "Success!\n");
return 0;
}
--
Gitblit v1.10.0