From 3a99b8151f6a002353691b1254b395eb52effefa Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 17 Aug 2015 17:25:43 +0000
Subject: [PATCH] Fair comparison
---
src/parser.c | 11 +++++++++--
1 files changed, 9 insertions(+), 2 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 2a82ee7..b373c01 100644
--- a/src/parser.c
+++ b/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)
{
--
Gitblit v1.10.0