src/layer.h
@@ -13,7 +13,10 @@ DROPOUT, CROP, ROUTE, COST COST, NORMALIZATION, REGION, AVGPOOL } LAYER_TYPE; typedef enum{ @@ -47,6 +50,13 @@ int objectness; int does_cost; int joint; int noadjust; float alpha; float beta; float kappa; int dontload; float probability; float scale; @@ -67,6 +77,8 @@ int * input_sizes; float * delta; float * output; float * squared; float * norms; #ifdef GPU int *indexes_gpu; @@ -84,6 +96,8 @@ float * output_gpu; float * delta_gpu; float * rand_gpu; float * squared_gpu; float * norms_gpu; #endif } layer;