From ff67f0347653c35c67ddbafad8dc76bbd868047e Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Wed, 03 Dec 2014 16:48:07 +0000 Subject: [PATCH] Starting on server/client --- src/network.h | 6 ++++++ 1 files changed, 6 insertions(+), 0 deletions(-) diff --git a/src/network.h b/src/network.h index 22e277c..eeaef41 100644 --- a/src/network.h +++ b/src/network.h @@ -42,6 +42,9 @@ cl_mem get_network_output_cl_layer(network net, int i); 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); @@ -50,10 +53,13 @@ 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); float network_accuracy(network net, data d); float network_accuracy_multi(network net, data d, int n); +void top_predictions(network net, int n, int *index); float *get_network_output(network net); float *get_network_output_layer(network net, int i); float *get_network_delta_layer(network net, int i); -- Gitblit v1.10.0