From b2b7137b6f185ce2f01664d782a09b08d50d5a07 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Tue, 28 Jan 2014 07:16:56 +0000 Subject: [PATCH] About to do something stupid... --- src/network.h | 3 ++- 1 files changed, 2 insertions(+), 1 deletions(-) diff --git a/src/network.h b/src/network.h index 2ffc76b..fa109dd 100644 --- a/src/network.h +++ b/src/network.h @@ -22,9 +22,10 @@ network make_network(int n); void forward_network(network net, double *input); -void backward_network(network net, double *input, double *truth); +double backward_network(network net, double *input, double *truth); void update_network(network net, double step, double momentum, double decay); double train_network_sgd(network net, data d, int n, double step, double momentum,double decay); +double train_network_batch(network net, data d, int n, double step, double momentum,double decay); void train_network(network net, data d, double step, double momentum, double decay); matrix network_predict_data(network net, data test); double network_accuracy(network net, data d); -- Gitblit v1.10.0