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

Fixed LSTM-layer

Browse Source
This commit is contained in:
AlexeyAB
2019-01-28 20:22:14 +03:00
parent 85b99872cb
commit 110b5240a4
14 changed files with 130 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/demo.c
View File

@ -152,7 +152,7 @@ void demo(char *cfgfile, char *weightfile, float thresh, float hier_thresh, int
demo_ext_output = ext_output;
demo_json_port = json_port;
printf("Demo\n");
net = parse_network_cfg_custom(cfgfile, 1); // set batch=1
net = parse_network_cfg_custom(cfgfile, 1, 0); // set batch=1
if(weightfile){
load_weights(&net, weightfile);
}