Joseph Redmon
2015-07-11 ccde487525fc89a1d4bc3e1cf11a18971e8451c9
src/layer.h
@@ -13,7 +13,8 @@
    DROPOUT,
    CROP,
    ROUTE,
    COST
    COST,
    NORMALIZATION
} LAYER_TYPE;
typedef enum{
@@ -48,6 +49,10 @@
    int does_cost;
    int joint;
    float alpha;
    float beta;
    float kappa;
    int dontload;
    float probability;
@@ -69,6 +74,8 @@
    int   * input_sizes;
    float * delta;
    float * output;
    float * squared;
    float * norms;
    #ifdef GPU
    int *indexes_gpu;
@@ -86,6 +93,8 @@
    float * output_gpu;
    float * delta_gpu;
    float * rand_gpu;
    float * squared_gpu;
    float * norms_gpu;
    #endif
} layer;