Minor fix

AlexeyAB

2018-05-27 160eddddc4e265d5ee59a38797c30720bf46cd7c

 src/layer.h

@@ -3,6 +3,7 @@

#include "activations.h"
#include "stddef.h"
#include "tree.h"

struct network_state;

@@ -32,7 +33,10 @@
    NETWORK,
    XNOR,
    REGION,
   YOLO,
    REORG,
   UPSAMPLE,
   REORG_OLD,
    BLANK
} LAYER_TYPE;

@@ -40,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;
@@ -66,6 +82,7 @@
    int size;
    int side;
    int stride;
    int reverse;
    int pad;
    int sqrt;
    int flip;
@@ -81,6 +98,7 @@
    float exposure;
    float shift;
    float ratio;
   int focal_loss;
    int softmax;
    int classes;
    int coords;
@@ -92,6 +110,23 @@
    int noadjust;
    int reorg;
    int log;
   int tanh;
   int *mask;
   int total;
   float bflops;

    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;
@@ -100,9 +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;

@@ -220,6 +265,9 @@
    float * weights_gpu;
    float * weight_updates_gpu;

   float * weights_gpu16;
   float * weight_updates_gpu16;

    float * biases_gpu;
    float * bias_updates_gpu;

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