src/layer.h
@@ -13,7 +13,9 @@ DROPOUT, CROP, ROUTE, COST COST, NORMALIZATION, AVGPOOL } LAYER_TYPE; typedef enum{ @@ -47,6 +49,11 @@ int objectness; int does_cost; int joint; int noadjust; float alpha; float beta; float kappa; int dontload; @@ -69,6 +76,8 @@ int * input_sizes; float * delta; float * output; float * squared; float * norms; #ifdef GPU int *indexes_gpu; @@ -86,6 +95,8 @@ float * output_gpu; float * delta_gpu; float * rand_gpu; float * squared_gpu; float * norms_gpu; #endif } layer;