From c2578bcce0d3e791a3afbe6e27700c19450d37eb Mon Sep 17 00:00:00 2001 From: LamHoCN <18795959682@163.com> Date: Sat, 4 Nov 2017 20:28:42 +0800 Subject: [PATCH] Update parser.c --- src/parser.c | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/parser.c b/src/parser.c index 1b257f2..04a14b4 100644 --- a/src/parser.c +++ b/src/parser.c @@ -793,6 +793,7 @@ network parse_network_cfg(char *filename) l.smooth = option_find_float_quiet(options, "smooth", 0); option_unused(options); net.layers[count] = l; + //这里就是为workspace 做最大化的操作 -merlin if (l.workspace_size > workspace_size) workspace_size = l.workspace_size; free_section(s); n = n->next; @@ -804,7 +805,7 @@ network parse_network_cfg(char *filename) params.c = l.out_c; params.inputs = l.outputs; } - } + } free_list(sections); layer out = get_network_output_layer(net); net.outputs = out.outputs; @@ -1244,7 +1245,7 @@ void load_weights_upto(network *net, char *filename, int start, int cutoff) fprintf(stderr, "Done!\n"); fclose(fp); } - +//从.weights中读取权重的操作 void load_weights(network *net, char *filename) { load_weights_upto(net, filename, 0, net->n);