src/layer.h
@@ -28,6 +28,8 @@ CRNN, BATCHNORM, NETWORK, XNOR, REGION, BLANK } LAYER_TYPE; @@ -87,6 +89,7 @@ float object_scale; float noobject_scale; float class_scale; int random; int dontload; int dontloadscales; @@ -102,6 +105,9 @@ char *cfilters; float *filter_updates; float *state; float *prev_state; float *forgot_state; float *forgot_delta; float *state_delta; float *concat; @@ -159,6 +165,12 @@ struct layer *input_h_layer; struct layer *state_h_layer; float *z_cpu; float *r_cpu; float *h_cpu; float *binary_input; size_t workspace_size; #ifdef GPU