From 1d53b6414e0cd81043d7c76aa89f4f97da5e479f Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Thu, 23 Jan 2014 19:24:37 +0000 Subject: [PATCH] Stable on MNIST, about to change a lot --- src/network.h | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/src/network.h b/src/network.h index 2ffc76b..975c3dd 100644 --- a/src/network.h +++ b/src/network.h @@ -25,6 +25,7 @@ void 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