blame | history | patch | commit | commitdiff | ignore whitespace
AlexeyAB
2018-03-19 ced198e9390195875d743d77eadece99c7fd5b38 
 src/layer.h


@@ -3,6 +3,9 @@




#include "activations.h"


#include "stddef.h"


#include "tree.h"




struct network_state;




struct layer;


typedef struct layer layer;


@@ -31,6 +34,7 @@


    XNOR,


    REGION,


    REORG,


   REORG_OLD,


    BLANK


} LAYER_TYPE;




@@ -38,10 +42,28 @@


    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;


    COST_TYPE cost_type;


    void (*forward)   (struct layer, struct network_state);


    void (*backward)  (struct layer, struct network_state);


    void (*update)    (struct layer, int, float, float, float);


    void (*forward_gpu)   (struct layer, struct network_state);


    void (*backward_gpu)  (struct layer, struct network_state);


    void (*update_gpu)    (struct layer, int, float, float, float);


    int batch_normalize;


    int shortcut;


    int batch;


@@ -54,10 +76,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;


@@ -73,6 +97,7 @@


    float exposure;


    float shift;


    float ratio;


   int focal_loss;


    int softmax;


    int classes;


    int coords;


@@ -85,6 +110,19 @@


    int reorg;


    int log;




    int adam;


    float B1;


    float B2;


    float eps;


    float *m_gpu;


    float *v_gpu;


    int t;


    float *m;


    float *v;




    tree *softmax_tree;


    int  *map;




    float alpha;


    float beta;


    float kappa;


@@ -93,8 +131,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;




@@ -212,6 +256,9 @@


    float * weights_gpu;


    float * weight_updates_gpu;




   float * weights_gpu16;


   float * weight_updates_gpu16;




    float * biases_gpu;


    float * bias_updates_gpu;




@@ -226,6 +273,7 @@


    #ifdef CUDNN


    cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;


    cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;


   cudnnTensorDescriptor_t normTensorDesc;


    cudnnFilterDescriptor_t weightDesc;


    cudnnFilterDescriptor_t dweightDesc;


    cudnnConvolutionDescriptor_t convDesc;