From 79fffcce3ce495bd415dc1284224c915d7194d4c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 11 Dec 2014 21:15:26 +0000
Subject: [PATCH] Better imagenet distributed training
---
src/network.c | 74 ++++++++++++++++++++++++++++++++++++-
1 files changed, 72 insertions(+), 2 deletions(-)
diff --git a/src/network.c b/src/network.c
index 3a6a184..ae030ce 100644
--- a/src/network.c
+++ b/src/network.c
@@ -323,6 +323,65 @@
fprintf(stderr, "Accuracy: %f\n", (float)correct/d.X.rows);
}
+void set_learning_network(network *net, float rate, float momentum, float decay)
+{
+ int i;
+ net->learning_rate=rate;
+ net->momentum = momentum;
+ net->decay = decay;
+ for(i = 0; i < net->n; ++i){
+ if(net->types[i] == CONVOLUTIONAL){
+ convolutional_layer *layer = (convolutional_layer *)net->layers[i];
+ layer->learning_rate=rate;
+ layer->momentum = momentum;
+ layer->decay = decay;
+ }
+ else if(net->types[i] == CONNECTED){
+ connected_layer *layer = (connected_layer *)net->layers[i];
+ layer->learning_rate=rate;
+ layer->momentum = momentum;
+ layer->decay = decay;
+ }
+ }
+}
+
+
+void set_batch_network(network *net, int b)
+{
+ net->batch = b;
+ int i;
+ for(i = 0; i < net->n; ++i){
+ if(net->types[i] == CONVOLUTIONAL){
+ convolutional_layer *layer = (convolutional_layer *)net->layers[i];
+ layer->batch = b;
+ }
+ else if(net->types[i] == MAXPOOL){
+ maxpool_layer *layer = (maxpool_layer *)net->layers[i];
+ layer->batch = b;
+ }
+ else if(net->types[i] == CONNECTED){
+ connected_layer *layer = (connected_layer *)net->layers[i];
+ layer->batch = b;
+ } else if(net->types[i] == DROPOUT){
+ dropout_layer *layer = (dropout_layer *) net->layers[i];
+ layer->batch = b;
+ }
+ else if(net->types[i] == FREEWEIGHT){
+ freeweight_layer *layer = (freeweight_layer *) net->layers[i];
+ layer->batch = b;
+ }
+ else if(net->types[i] == SOFTMAX){
+ softmax_layer *layer = (softmax_layer *)net->layers[i];
+ layer->batch = b;
+ }
+ else if(net->types[i] == COST){
+ cost_layer *layer = (cost_layer *)net->layers[i];
+ layer->batch = b;
+ }
+ }
+}
+
+
int get_network_input_size_layer(network net, int i)
{
if(net.types[i] == CONVOLUTIONAL){
@@ -586,15 +645,26 @@
float network_accuracy(network net, data d)
{
matrix guess = network_predict_data(net, d);
- float acc = matrix_accuracy(d.y, guess);
+ float acc = matrix_topk_accuracy(d.y, guess,1);
free_matrix(guess);
return acc;
}
+float *network_accuracies(network net, data d)
+{
+ static float acc[2];
+ matrix guess = network_predict_data(net, d);
+ acc[0] = matrix_topk_accuracy(d.y, guess,1);
+ acc[1] = matrix_topk_accuracy(d.y, guess,5);
+ free_matrix(guess);
+ return acc;
+}
+
+
float network_accuracy_multi(network net, data d, int n)
{
matrix guess = network_predict_data_multi(net, d, n);
- float acc = matrix_accuracy(d.y, guess);
+ float acc = matrix_topk_accuracy(d.y, guess,1);
free_matrix(guess);
return acc;
}
--
Gitblit v1.10.0