From b13ad6d5fd23f68f506c14ede4282126d893702b Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 05 Nov 2014 22:49:58 +0000
Subject: [PATCH] Can validate on imagenet now

---
 src/network.h |    3 +++
 1 files changed, 3 insertions(+), 0 deletions(-)

diff --git a/src/network.h b/src/network.h
index 7625904..eeaef41 100644
--- a/src/network.h
+++ b/src/network.h
@@ -43,6 +43,8 @@
 cl_mem get_network_delta_cl_layer(network net, int i);
 float train_network_sgd_gpu(network net, data d, int n);
 float train_network_data_gpu(network net, data d, int n);
+float *network_predict_gpu(network net, float *input);
+float network_accuracy_gpu(network net, data d);
 #endif
 
 network make_network(int n, int batch);
@@ -51,6 +53,7 @@
 void update_network(network net);
 float train_network_sgd(network net, data d, int n);
 float train_network_batch(network net, data d, int n);
+float train_network_data_cpu(network net, data d, int n);
 void train_network(network net, data d);
 matrix network_predict_data(network net, data test);
 float *network_predict(network net, float *input);

--
Gitblit v1.10.0