AlexeyAB
2018-08-07 e6c97a53a7b5ac4014d30d236ea2bf5adb4bb521
src/layer.h
@@ -33,7 +33,9 @@
    NETWORK,
    XNOR,
    REGION,
   YOLO,
    REORG,
   UPSAMPLE,
   REORG_OLD,
    BLANK
} LAYER_TYPE;
@@ -42,6 +44,18 @@
    SSE, MASKED, SMOOTH
} COST_TYPE;
typedef struct {
   int batch;
   float learning_rate;
   float momentum;
   float decay;
   int adam;
   float B1;
   float B2;
   float eps;
   int t;
} update_args;
struct layer{
    LAYER_TYPE type;
    ACTIVATION activation;
@@ -64,7 +78,6 @@
    int out_h, out_w, out_c;
    int n;
    int max_boxes;
   int small_object;
    int groups;
    int size;
    int side;
@@ -97,6 +110,10 @@
    int noadjust;
    int reorg;
    int log;
   int tanh;
   int *mask;
   int total;
   float bflops;
    int adam;
    float B1;
@@ -118,10 +135,14 @@
    float coord_scale;
    float object_scale;
    float noobject_scale;
   float mask_scale;
    float class_scale;
    int bias_match;
    int random;
   float ignore_thresh;
   float truth_thresh;
    float thresh;
   float focus;
    int classfix;
    int absolute;
@@ -261,6 +282,7 @@
    #ifdef CUDNN
    cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
    cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;
   cudnnTensorDescriptor_t normTensorDesc, normDstTensorDesc, normDstTensorDescF16;
    cudnnFilterDescriptor_t weightDesc;
    cudnnFilterDescriptor_t dweightDesc;
    cudnnConvolutionDescriptor_t convDesc;