Better partial function

Joseph Redmon

2015-07-20 38bd6ae6ba24fc8c14fd61d1238ae94a983434b3

Better partial function

 src/darknet.c

@@ -36,7 +36,7 @@
        load_weights_upto(&net, weightfile, max);
    }
    net.seen = 0;
    save_weights(net, outfile);
    save_weights_upto(net, outfile, max);
}

#include "convolutional_layer.h"

 src/parser.c

@@ -500,7 +500,7 @@
    return sections;
}

void save_weights(network net, char *filename)
void save_weights_upto(network net, char *filename, int cutoff)
{
    fprintf(stderr, "Saving weights to %s\n", filename);
    FILE *fp = fopen(filename, "w");
@@ -512,7 +512,7 @@
    fwrite(&net.seen, sizeof(int), 1, fp);

    int i;
    for(i = 0; i < net.n; ++i){
    for(i = 0; i < net.n && i < cutoff; ++i){
        layer l = net.layers[i];
        if(l.type == CONVOLUTIONAL){
#ifdef GPU
@@ -546,6 +546,10 @@
    }
    fclose(fp);
}
void save_weights(network net, char *filename)
{
    save_weights_upto(net, filename, net.n);
}

void load_weights_upto(network *net, char *filename, int cutoff)
{

 src/parser.h

@@ -5,6 +5,7 @@
network parse_network_cfg(char *filename);
void save_network(network net, char *filename);
void save_weights(network net, char *filename);
void save_weights_upto(network net, char *filename, int cutoff);
void load_weights(network *net, char *filename);
void load_weights_upto(network *net, char *filename, int cutoff);