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

Optimized memory allocation for Detection (inference only), without allocation memory for training

Browse Source
This commit is contained in:
AlexeyAB
2019-11-09 22:48:18 +03:00
parent 0eee8404bf
commit d91d59a22f
17 changed files with 170 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
include/darknet.h
View File

@ -190,6 +190,7 @@ struct layer {
void(*backward_gpu) (struct layer, struct network_state);
void(*update_gpu) (struct layer, int, float, float, float);
layer *share_layer;
int train;
int batch_normalize;
int shortcut;
int batch;