From a360f694093f7748e3232f4ed74d40446d735fc3 Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Wed, 01 Mar 2017 12:29:50 +0000
Subject: [PATCH] Readme.md - When should I stop training
---
src/parser.c | 38 ++++++++++++++++++++------------------
1 files changed, 20 insertions(+), 18 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index cde06b4..3f39a13 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -238,9 +238,6 @@
int classes = option_find_int(options, "classes", 20);
int num = option_find_int(options, "num", 1);
- params.w = option_find_int(options, "side", params.w);
- params.h = option_find_int(options, "side", params.h);
-
layer l = make_region_layer(params.batch, params.w, params.h, num, classes, coords);
assert(l.outputs == params.inputs);
@@ -829,7 +826,7 @@
}
#endif
fprintf(stderr, "Saving weights to %s\n", filename);
- FILE *fp = fopen(filename, "w");
+ FILE *fp = fopen(filename, "wb");
if(!fp) file_error(filename);
int major = 0;
@@ -969,23 +966,28 @@
//return;
}
int num = l.n*l.c*l.size*l.size;
- if(0){
- fread(l.biases + ((l.n != 1374)?0:5), sizeof(float), l.n, fp);
- if (l.batch_normalize && (!l.dontloadscales)){
- fread(l.scales + ((l.n != 1374)?0:5), sizeof(float), l.n, fp);
- fread(l.rolling_mean + ((l.n != 1374)?0:5), sizeof(float), l.n, fp);
- fread(l.rolling_variance + ((l.n != 1374)?0:5), sizeof(float), l.n, fp);
+ fread(l.biases, sizeof(float), l.n, fp);
+ if (l.batch_normalize && (!l.dontloadscales)){
+ fread(l.scales, sizeof(float), l.n, fp);
+ fread(l.rolling_mean, sizeof(float), l.n, fp);
+ fread(l.rolling_variance, sizeof(float), l.n, fp);
+ if(0){
+ int i;
+ for(i = 0; i < l.n; ++i){
+ printf("%g, ", l.rolling_mean[i]);
+ }
+ printf("\n");
+ for(i = 0; i < l.n; ++i){
+ printf("%g, ", l.rolling_variance[i]);
+ }
+ printf("\n");
}
- fread(l.weights + ((l.n != 1374)?0:5*l.c*l.size*l.size), sizeof(float), num, fp);
- }else{
- fread(l.biases, sizeof(float), l.n, fp);
- if (l.batch_normalize && (!l.dontloadscales)){
- fread(l.scales, sizeof(float), l.n, fp);
- fread(l.rolling_mean, sizeof(float), l.n, fp);
- fread(l.rolling_variance, sizeof(float), l.n, fp);
+ if(0){
+ fill_cpu(l.n, 0, l.rolling_mean, 1);
+ fill_cpu(l.n, 0, l.rolling_variance, 1);
}
- fread(l.weights, sizeof(float), num, fp);
}
+ fread(l.weights, sizeof(float), num, fp);
if(l.adam){
fread(l.m, sizeof(float), num, fp);
fread(l.v, sizeof(float), num, fp);
--
Gitblit v1.10.0