mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Better load messaging

Browse Source
This commit is contained in:
Joseph Redmon
2015-06-10 11:14:04 -07:00
parent ae43771ca2
commit afe245329b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/parser.c
View File

@@ -514,7 +514,8 @@ void save_weights(network net, char *filename)
void load_weights_upto(network *net, char *filename, int cutoff) void load_weights_upto(network *net, char *filename, int cutoff)
{ {
fprintf(stderr, "Loading weights from %s\n", filename); fprintf(stderr, "Loading weights from %s...", filename);
fflush(stdout);
FILE *fp = fopen(filename, "r"); FILE *fp = fopen(filename, "r");
if(!fp) file_error(filename); if(!fp) file_error(filename);
@@ -522,7 +523,6 @@ void load_weights_upto(network *net, char *filename, int cutoff)
fread(&net->momentum, sizeof(float), 1, fp); fread(&net->momentum, sizeof(float), 1, fp);
fread(&net->decay, sizeof(float), 1, fp); fread(&net->decay, sizeof(float), 1, fp);
fread(&net->seen, sizeof(int), 1, fp); fread(&net->seen, sizeof(int), 1, fp);
fprintf(stderr, "%f %f %f %d\n", net->learning_rate, net->momentum, net->decay, net->seen);
int i; int i;
for(i = 0; i < net->n && i < cutoff; ++i){ for(i = 0; i < net->n && i < cutoff; ++i){
@@ -557,6 +557,7 @@ void load_weights_upto(network *net, char *filename, int cutoff)
#endif #endif
} }
} }
fprintf(stderr, "Done!\n");
fclose(fp); fclose(fp);
} }