Joseph Redmon
2015-08-17 741ada451cc7fee1b9a4c3deaec6af87a2af7497
src/parser.c
@@ -236,6 +236,9 @@
{
    float probability = option_find_float(options, "probability", .5);
    dropout_layer layer = make_dropout_layer(params.batch, params.inputs, probability);
    layer.out_w = params.w;
    layer.out_h = params.h;
    layer.out_c = params.c;
    return layer;
}
@@ -500,7 +503,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 +515,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 +549,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)
{