AlexeyAB
2018-03-08 a6c51e3b758aee7fd3a6f1d37daa8dcad4891e52
src/layer.h
@@ -34,6 +34,7 @@
    XNOR,
    REGION,
    REORG,
   REORG_OLD,
    BLANK
} LAYER_TYPE;
@@ -63,6 +64,7 @@
    int out_h, out_w, out_c;
    int n;
    int max_boxes;
   int small_object;
    int groups;
    int size;
    int side;
@@ -83,6 +85,7 @@
    float exposure;
    float shift;
    float ratio;
   int focal_loss;
    int softmax;
    int classes;
    int coords;
@@ -241,6 +244,9 @@
    float * weights_gpu;
    float * weight_updates_gpu;
   float * weights_gpu16;
   float * weight_updates_gpu16;
    float * biases_gpu;
    float * bias_updates_gpu;