From a08ef29e085236f13fd04246c6774f9682a56208 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 10 Jul 2015 23:38:14 +0000
Subject: [PATCH] NIGHTMARE!!!!
---
src/parser.c | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 2caf96e..18c3860 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -343,6 +343,7 @@
}else{
fprintf(stderr, "Type not recognized: %s\n", s->type);
}
+ l.dontload = option_find_int_quiet(options, "dontload", 0);
net.layers[count] = l;
free_section(s);
n = n->next;
@@ -527,6 +528,7 @@
int i;
for(i = 0; i < net->n && i < cutoff; ++i){
layer l = net->layers[i];
+ if (l.dontload) continue;
if(l.type == CONVOLUTIONAL){
int num = l.n*l.c*l.size*l.size;
fread(l.biases, sizeof(float), l.n, fp);
--
Gitblit v1.10.0