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

Fix training approach (convolutional layer)

Browse Source
This commit is contained in:
AlexeyAB
2018-12-27 00:31:28 +03:00
parent dc827f4c1c
commit 64e478db07
4 changed files with 68 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/parser.c
View File

@ -805,6 +805,7 @@ network parse_network_cfg_custom(char *filename, int batch)
l.stopbackward = option_find_int_quiet(options, "stopbackward", 0);
l.dontload = option_find_int_quiet(options, "dontload", 0);
l.dontloadscales = option_find_int_quiet(options, "dontloadscales", 0);
l.learning_rate_scale = option_find_float_quiet(options, "learning_rate", 1);
option_unused(options);
net.layers[count] = l;
if (l.workspace_size > workspace_size) workspace_size = l.workspace_size;