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

add SGDR policy

Browse Source
This commit is contained in:
Josh Veitch-Michaelis
2019-03-18 23:26:04 +00:00
parent 8bcba6c105
commit d64693eb77
3 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/parser.c
View File

@ -629,6 +629,7 @@ learning_rate_policy get_policy(char *s)
if (strcmp(s, "exp")==0) return EXP;
if (strcmp(s, "sigmoid")==0) return SIG;
if (strcmp(s, "steps")==0) return STEPS;
if (strcmp(s, "sgdr")==0) return SGDR;
fprintf(stderr, "Couldn't find policy %s, going with constant\n", s);
return CONSTANT;
}
@ -637,6 +638,9 @@ void parse_net_options(list *options, network *net)
{
net->batch = option_find_int(options, "batch",1);
net->learning_rate = option_find_float(options, "learning_rate", .001);
net->learning_rate_min = option_find_float_quiet(options, "learning_rate_min", .00001);
net->learning_rate_max = option_find_float_quiet(options, "learning_rate_max", .001);
net->batches_per_cycle = option_find_int(options, "sgdr_cycle", 500);
net->momentum = option_find_float(options, "momentum", .9);
net->decay = option_find_float(options, "decay", .0001);
int subdivs = option_find_int(options, "subdivisions",1);