AlexeyAB
2018-03-05 ae1768e5831caa95214b93b08ee711aede36df07
src/layer.h
@@ -34,6 +34,7 @@
    XNOR,
    REGION,
    REORG,
   REORG_OLD,
    BLANK
} LAYER_TYPE;
@@ -63,10 +64,12 @@
    int out_h, out_w, out_c;
    int n;
    int max_boxes;
   int small_object;
    int groups;
    int size;
    int side;
    int stride;
    int reverse;
    int pad;
    int sqrt;
    int flip;
@@ -82,6 +85,7 @@
    float exposure;
    float shift;
    float ratio;
   int focal_loss;
    int softmax;
    int classes;
    int coords;
@@ -101,8 +105,11 @@
    float *m_gpu;
    float *v_gpu;
    int t;
    float *m;
    float *v;
    tree *softmax_tree;
    int  *map;
    float alpha;
    float beta;
@@ -112,8 +119,14 @@
    float object_scale;
    float noobject_scale;
    float class_scale;
    int bias_match;
    int random;
    float thresh;
    int classfix;
    int absolute;
    int onlyforward;
    int stopbackward;
    int dontload;
    int dontloadscales;
@@ -231,6 +244,9 @@
    float * weights_gpu;
    float * weight_updates_gpu;
   float * weights_gpu16;
   float * weight_updates_gpu16;
    float * biases_gpu;
    float * bias_updates_gpu;