blame | history | patch | commit | commitdiff | ignore whitespace
AlexeyAB
2018-08-07 0a326e7afe3e690c0b4cb64bbd0ce4f0603a7d85 
 src/layer.h


@@ -33,7 +33,10 @@


    NETWORK,


    XNOR,


    REGION,


   YOLO,


    REORG,


   UPSAMPLE,


   REORG_OLD,


    BLANK


} LAYER_TYPE;




@@ -41,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;


@@ -67,6 +82,7 @@


    int size;


    int side;


    int stride;


    int reverse;


    int pad;


    int sqrt;


    int flip;


@@ -82,6 +98,7 @@


    float exposure;


    float shift;


    float ratio;


   int focal_loss;


    int softmax;


    int classes;


    int coords;


@@ -93,6 +110,10 @@


    int noadjust;


    int reorg;


    int log;


   int tanh;


   int *mask;


   int total;


   float bflops;




    int adam;


    float B1;


@@ -114,11 +135,19 @@


    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;




    int onlyforward;


    int stopbackward;


    int dontload;


    int dontloadscales;




@@ -150,6 +179,9 @@


    float *weights;


    float *weight_updates;




    char *align_bit_weights;


    float *mean_arr;




    float *col_image;


    int   * input_layers;


    int   * input_sizes;


@@ -236,6 +268,9 @@


    float * weights_gpu;


    float * weight_updates_gpu;




   float * weights_gpu16;


   float * weight_updates_gpu16;




    float * biases_gpu;


    float * bias_updates_gpu;




@@ -250,6 +285,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;