From ace5aeb0f59fdceb99e607af9780added20da37c Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Fri, 24 Jan 2014 22:51:17 +0000 Subject: [PATCH] MNIST connected network showing off matrices --- 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